pep8

williamFalcon · williamFalcon · commit e7ea564df2bf · 2020-04-27T08:47:19.000-04:00
diff --git a/tests/loggers/test_all.py b/tests/loggers/test_all.py
@@ -77,8 +77,6 @@ def log_metrics(self, metrics, step):
     # 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()
 
diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
@@ -24,7 +24,6 @@
     LightTestDataloader,
     LightValidationMixin,
 )
-from tests.base import TestModelBase
 
 
 def test_model_pickle(tmpdir):

Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,6 @@`
`24`	`24`	`LightTestDataloader,`
`25`	`25`	`LightValidationMixin,`
`26`	`26`	`)`
`27`		`-from tests.base import TestModelBase`
`28`	`27`
`29`	`28`
`30`	`29`	`def test_model_pickle(tmpdir):`