diff --git a/tests/models/test_tpu.py b/tests/models/test_tpu.py index 207c97abc735df..0554d924e6e9fb 100644 --- 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):