...

Author: Borda

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):