From e7ea564df2bf11e8efd7a7853bed067781e7d487 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Mon, 27 Apr 2020 08:47:19 -0400
Subject: [PATCH] pep8

---
 tests/loggers/test_all.py     | 2 --
 tests/trainer/test_trainer.py | 1 -
 2 files changed, 3 deletions(-)

diff --git a/tests/loggers/test_all.py b/tests/loggers/test_all.py
index 85751204..383ca263 100644
--- a/tests/loggers/test_all.py
+++ b/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()
 
diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
index 116cafcc..18cc2586 100644
--- a/tests/trainer/test_trainer.py
+++ b/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):