Skip to content

Commit

Permalink
disable testOp when numpy 2.0 is installed
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 672712350
  • Loading branch information
trax-robot authored and copybara-github committed Sep 10, 2024
1 parent 39c9bbb commit e100df7
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions trax/tf_numpy/jax_tests/lax_numpy_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit e100df7

Please sign in to comment.