From 40015ddd1030bff70f32a2ab7b1f498a68e83ffb Mon Sep 17 00:00:00 2001 From: tchaton Date: Mon, 29 Mar 2021 17:18:54 +0100 Subject: [PATCH] update --- tests/utilities/test_xla_device_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/utilities/test_xla_device_utils.py b/tests/utilities/test_xla_device_utils.py index 46fe6403c93ae..edca2777b578a 100644 --- 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)