...

Lightning-AI · Mar 23, 2021 · dd01e03 · dd01e03
1 parent 9ce794c
commit dd01e03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/test_tpu.py b/tests/models/test_tpu.py
@@ -350,7 +350,7 @@ def test_reduce(rank):
 
 
 @pytest.mark.parametrize("clip_val", [0, 10])
-@RunIf(tpu=True)
+@pytest.mark.skipif(not _TPU_AVAILABLE, reason="test requires TPU machine")
 @pl_multi_process_test
 @mock.patch("pytorch_lightning.accelerators.tpu.xla_clip_grad_norm_")
 def test_tpu_precision_16_clip_gradients(mock_clip_grad_norm, clip_val, tmpdir):