diff --git a/tests/lax_numpy_test.py b/tests/lax_numpy_test.py index f93e28dada71..ddf42a28e2ba 100644 --- a/tests/lax_numpy_test.py +++ b/tests/lax_numpy_test.py @@ -2134,6 +2134,9 @@ def np_fun(x): self._CheckAgainstNumpy(np_fun, jnp_fun, args_maker) @jtu.sample_product(dtype=inexact_dtypes, equal_nan=[True, False]) + @jtu.ignore_warning( + category=RuntimeWarning, message='invalid value encountered in cast' + ) def testUniqueEqualNan(self, dtype, equal_nan): shape = (20,) rng = jtu.rand_some_nan(self.rng())