Merge pull request #20665 from jakevdp:fix-clip-test

jax authors · jax authors · commit 675fdba2d645 · 2024-04-09T10:37:33.000-07:00
PiperOrigin-RevId: 623213147
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -862,7 +862,7 @@ def testPolyfit(self, shape, dtype, deg, rcond, full, w, cov):`
`862`	`862`	`(np.full(1, -0.9), np.ones(1))]`
`863`	`863`	`],`
`864`	`864`	`shape=all_shapes,`
`865`		`- dtype=number_dtypes,`
	`865`	`+ dtype=float_dtypes + int_dtypes + unsigned_dtypes,`
`866`	`866`	`)`
`867`	`867`	`@jax.numpy_rank_promotion('allow') # This test explicitly exercises implicit rank promotion.`
`868`	`868`	`@jax.numpy_dtype_promotion('standard') # This test explicitly exercises mixed type promotion`