diff --git a/tests/accelerators/test_tpu.py b/tests/accelerators/test_tpu.py index 58fcca9ac5249a..5fc2aba8cbe575 100644 --- a/tests/accelerators/test_tpu.py +++ b/tests/accelerators/test_tpu.py @@ -313,6 +313,7 @@ def test_warning_if_tpus_not_used(): Trainer() +@pytest.mark.skip(reason="TODO(@kaushikb11): Optimize TPU tests to avoid timeouts") @RunIf(tpu=True) @pytest.mark.parametrize( ["devices", "expected_device_ids"],