added base tests for tpu

Lightning-AI · Jul 6, 2020 · 615a0c6 · 615a0c6
1 parent 0ff6c71
commit 615a0c6
Showing 1 changed file with 1 addition and 1 deletion.
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,