disable testOp when numpy 2.0 is installed

PiperOrigin-RevId: 672712350
google · Sep 10, 2024 · d93cf1b · d93cf1b
1 parent 17c19c7
commit d93cf1b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/trax/tf_numpy/jax_tests/lax_numpy_test.py b/trax/tf_numpy/jax_tests/lax_numpy_test.py
@@ -500,6 +500,8 @@ def f():
           *(_valid_dtypes_for_shape(s, rec.dtypes) for s in shapes)))
       for rec in itertools.chain(JAX_ONE_TO_ONE_OP_RECORDS,
                                  JAX_COMPOUND_OP_RECORDS)))
+  @unittest.skipIf(onp.__version__ >= onp.lib.NumpyVersion('2.0.0'),
+                   'tf numpy is implemented to be numpy 1.x compatible')
   def testOp(self, onp_op, lnp_op, rng_factory, shapes, dtypes, check_dtypes,
              tolerance, inexact, check_incomplete_shape):
     # TODO(b/147769803): Remove this skipping