Disable TF tests on macos

.github/workflows/macos.yml

@@ -24,7 +24,7 @@ jobs:
         python-version: ['3.10', 3.11, 3.12]
         JAX_ENABLE_X64: [0]
 
-    runs-on: macos-14
+    runs-on: macos-latest
 
     steps:
 

tests/experimental/empirical_tf_test.py

@@ -14,6 +14,7 @@
 
 """Tests for `experimental/empirical_tf/empirical.py`."""
 
+import platform
 from absl.testing import absltest
 from absl.testing import parameterized
 import jax
@@ -285,6 +286,8 @@ def test_keras_functional(
       diagonal_axes,
       vmap_axes,
   ):
+    if platform.system() == 'Darwin':
+      self.skipTest('TF <-> JAX fails on MacOS.')
     with jax.numpy_rank_promotion("warn"):
       f = f(classes=1, input_shape=input_shape, weights=None)
       f.build((None, *input_shape))
@@ -314,6 +317,8 @@ def test_keras_sequential(
       diagonal_axes,
       vmap_axes,
   ):
+    if platform.system() == 'Darwin':
+      self.skipTest('TF <-> JAX fails on MacOS.')
     with jax.numpy_rank_promotion("warn"):
       f = keras.Sequential()
       f.add(keras.layers.Conv2D(4, (3, 3), activation='relu'))
@@ -355,6 +360,8 @@ def test_tf_function(
       diagonal_axes,
       vmap_axes,
   ):
+    if platform.system() == 'Darwin':
+      self.skipTest('TF <-> JAX fails on MacOS.')
     f, f_jax = f_f_jax
     f = tf.function(f, input_signature=_input_signature)
     params = tf.random.normal(params_shape, seed=4)
@@ -379,6 +386,8 @@ def test_tf_module(
       diagonal_axes,
       vmap_axes,
   ):
+    if platform.system() == 'Darwin':
+      self.skipTest('TF <-> JAX fails on MacOS.')
     f = _MLP(input_size=5, sizes=[4, 6, 3], name='MLP')
     f_jax, params = experimental.get_apply_fn_and_params(f)
     self._compare_ntks(f, f_jax, params, trace_axes, diagonal_axes, vmap_axes)