diff --git a/tests/pallas/tpu_sparsecore_pallas_debug_check_test.py b/tests/pallas/tpu_sparsecore_pallas_debug_check_test.py index fb95c997c636..55cb1ce37c01 100644 --- a/tests/pallas/tpu_sparsecore_pallas_debug_check_test.py +++ b/tests/pallas/tpu_sparsecore_pallas_debug_check_test.py @@ -68,9 +68,9 @@ def setUp(self): super().setUp() def test_scalar_debug_check(self): - if not jtu.is_device_tpu_at_least(6): - # TODO: b/436509694 - Figure out why the test gets stuck on v5p. - self.skipTest("") + if not jtu.is_device_tpu_at_least(7): + # TODO: b/469486032 - Figure out why the test gets stuck on v5p, v6e. + self.skipTest("Fails on v5p and v6e.") x = jnp.arange(8)