update test

Lightning-AI · Apr 29, 2024 · 1c7d7c8 · 1c7d7c8
1 parent d194976
commit 1c7d7c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/tests_pytorch/accelerators/test_xla.py b/tests/tests_pytorch/accelerators/test_xla.py
@@ -56,7 +56,7 @@ def test_resume_training_on_cpu(tmp_path):
     """Checks if training can be resumed from a saved checkpoint on CPU."""
     # Train a model on TPU
     model = BoringModel()
-    trainer = Trainer(max_epochs=1, accelerator="tpu", devices="auto")
+    trainer = Trainer(max_epochs=1, accelerator="tpu", devices="auto", default_root_dir=tmp_path)
     trainer.fit(model)
 
     if trainer.world_size != trainer.num_devices: