added base tests for tpu

Lightning-AI · Jul 5, 2020 · a4a0a2f · a4a0a2f
1 parent 37ed69e
commit a4a0a2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/loggers/test_all.py b/tests/loggers/test_all.py
@@ -94,7 +94,7 @@ def test_loggers_pickle(tmpdir, monkeypatch, logger_class):
     logger = logger_class(**logger_args)
 
     # test pickling loggers
-    pickle.dumps(logger)
+    pickle.dumps(logger.experiment)
 
     trainer = Trainer(
         max_epochs=1,