removes previously introduced jax fix, now fixed upstream

LuzGarciaP · Jul 26, 2020 · 5fc49b6 · 5fc49b6
1 parent 95da132
commit 5fc49b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax_cosmo/scipy/interpolate.py b/jax_cosmo/scipy/interpolate.py
@@ -28,7 +28,7 @@ def interp(x, xp, fp):
     ind = np.argmin((x - xp) ** 2)
 
     # Perform linear interpolation
-    ind = np.clip(ind.astype("float32"), 1, len(xp) - 2).astype("int32")
+    ind = np.clip(ind, 1, len(xp) - 2)
 
     xi = xp[ind]
     # Figure out if we are on the right or the left of nearest