pep8

tests/loggers/test_all.py

@@ -77,8 +77,6 @@ def test_loggers_fit_test(tmpdir, monkeypatch, logger_class):
     # WandbLogger,  # TODO: add this one
 ])
 def test_loggers_pickle(tmpdir, monkeypatch, logger_class):
-    import pickle
-
     """Verify that pickling trainer with logger works."""
     tutils.reset_seed()
 

tests/trainer/test_trainer.py

@@ -24,7 +24,6 @@ from tests.base import (
     LightTestDataloader,
     LightValidationMixin,
 )
-from tests.base import TestModelBase
 
 
 def test_model_pickle(tmpdir):