Merge pull request #187 from jakevdp:test-fix

PiperOrigin-RevId: 673974000
jax-ml · Sep 12, 2024 · b39b73c · b39b73c
2 parents 4893435 + 74daa1a
commit b39b73c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ml_dtypes/tests/custom_float_test.py b/ml_dtypes/tests/custom_float_test.py
@@ -881,6 +881,7 @@ def testBinaryPredicateUfunc(self, float_type):
  op(x, y), op(x.astype(np.float32), y.astype(np.float32))
  )
 
+ @ignore_warning(category=RuntimeWarning, message="invalid value encountered")
  def testPredicateUfunc(self, float_type):
  for op in [np.isfinite, np.isinf, np.isnan, np.signbit, np.logical_not]:
  with self.subTest(op.__name__):