diff --git a/ml_dtypes/tests/custom_float_test.py b/ml_dtypes/tests/custom_float_test.py index a69c09a4..6437e4d3 100644 --- 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__):