update

Lightning-AI · Mar 29, 2021 · 40015dd · 40015dd
1 parent d178e1d
commit 40015dd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/utilities/test_xla_device_utils.py b/tests/utilities/test_xla_device_utils.py
@@ -39,6 +39,7 @@ def sleep_fn(sleep_time: float) -> bool:
 
 
 @patch('pytorch_lightning.utilities.xla_device.TPU_CHECK_TIMEOUT', 3)
+@pytest.mark.skipif(not _XLA_AVAILABLE, reason="test requires torch_xla to be present")
 def test_result_returns_within_timeout_seconds():
     """Check that pl_multi_process returns within 3 seconds"""
     fn = xla_utils.pl_multi_process(sleep_fn)