From f6851397acd3779074aedbf1c2c5ae6a425b81af Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Tue, 9 Apr 2024 09:13:26 -0700 Subject: [PATCH] test: fix testClipStaticBounds complex warning --- tests/lax_numpy_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/lax_numpy_test.py b/tests/lax_numpy_test.py index 30109fd91a26..96232fd87082 100644 --- a/tests/lax_numpy_test.py +++ b/tests/lax_numpy_test.py @@ -862,7 +862,7 @@ def testPolyfit(self, shape, dtype, deg, rcond, full, w, cov): (np.full(1, -0.9), np.ones(1))] ], shape=all_shapes, - dtype=number_dtypes, + dtype=float_dtypes + int_dtypes + unsigned_dtypes, ) @jax.numpy_rank_promotion('allow') # This test explicitly exercises implicit rank promotion. @jax.numpy_dtype_promotion('standard') # This test explicitly exercises mixed type promotion