Fix polydiv kokoro tests

google · Apr 13, 2022 · 34e206a · 34e206a
1 parent 191c838
commit 34e206a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/lax_numpy_test.py b/tests/lax_numpy_test.py
@@ -2885,6 +2885,8 @@ def testPolyMul(self, a1_shape, a2_shape, dtype):
   def testPolyDiv(self, a_shape, b_shape, dtype):
     rng = jtu.rand_default(self.rng())
 
+    @jtu.ignore_warning(category=RuntimeWarning, message="divide by zero.*")
+    @jtu.ignore_warning(category=RuntimeWarning, message="invalid value.*")
     def np_fun(arg1, arg2):
       q, r = np.polydiv(arg1, arg2)
       while r.size < max(arg1.size, arg2.size): # Pad residual to same size