skip test_vit_b16 until jax-ml/jax#25066 is resolved

samuela · Nov 22, 2024 · 9e3ed38 · 9e3ed38
1 parent 93ed706
commit 9e3ed38
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_all_the_things.py b/tests/test_all_the_things.py
@@ -430,6 +430,7 @@ def test_resnet18():
   # to reasonable expectations.
 
 
+@pytest.mark.skip(reason="https://github.com/jax-ml/jax/issues/25066")
 @pytest.mark.skipif(not is_network_reachable(), reason="Network is not reachable")
 def test_vit_b16():
   import torchvision