From 15a19969de5d37508c893d0ff5ccaf3d179a81bd Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Fri, 17 Jun 2022 16:52:22 -0700 Subject: [PATCH] [x64] make polynomial_test compatible with strict dtype promotion --- jax/_src/numpy/polynomial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jax/_src/numpy/polynomial.py b/jax/_src/numpy/polynomial.py index 71ad4b01c..9279a8908 100644 --- a/jax/_src/numpy/polynomial.py +++ b/jax/_src/numpy/polynomial.py @@ -97,7 +97,7 @@ def roots(p, *, strip_zeros=True): else: roots = _roots_no_zeros(p) # combine roots and zero roots - roots = hstack((roots, zeros(trailing_zeros, p.dtype))) + roots = hstack((roots, zeros(trailing_zeros, roots.dtype))) return roots