fix tpu tests

Lightning-AI · Jun 30, 2020 · 04e68f0 · 04e68f0
1 parent fc26078
commit 04e68f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -1193,7 +1193,7 @@ def test(
             # attempt to load weights from a spawn
             path = os.path.join(self.default_root_dir, '__temp_weight_ddp_end.ckpt')
             test_model = self.model
-            if os.path.exists(path):
+            if os.path.exists(path) and self.on_colab_kaggle:
                 test_model = self.load_spawn_weights(self.model)
 
             self.fit(test_model)