added base tests for tpu

williamFalcon · williamFalcon · commit 615a0c6472e7 · 2020-07-06T07:09:49.000-04:00
diff --git a/tests/loggers/test_all.py b/tests/loggers/test_all.py
@@ -98,7 +98,7 @@ def test_loggers_pickle(tmpdir, monkeypatch, logger_class):
     _ = logger.experiment
 
     # test pickling loggers
-    pickle.dumps(logger)
+    pickle.dumps(logger.experiment)
 
     trainer = Trainer(
         max_epochs=1,
