revert lazywhere hack

Author: lithomas1

scipy/_lib/_util.py

@@ -127,10 +127,6 @@ def _lazywhere(cond, arrays, f, fillvalue=None, f2=None):
     """
     xp = array_namespace(cond, *arrays)
 
-    if is_dask(xp) or is_jax(xp):
-        # TODO: verify for jax
-        return xp.where(cond, f(arrays[0], arrays[1]), f2(arrays[0], arrays[1]) if not fillvalue else fillvalue)
-
     if (f2 is fillvalue is None) or (f2 is not None and fillvalue is not None):
         raise ValueError("Exactly one of `fillvalue` or `f2` must be given.")
 

scipy/_lib/tests/test__util.py

@@ -360,7 +360,7 @@ def test_array_api(self, xp, nan_policy):
         x = xp.asarray(x0)
         # Non array-api-compat dask.asarray doesn't copy correctly
         # so do this instead of xp_copy
-        x_nan = xp.asarray(x.copy())
+        x_nan = xp.asarray(x0.copy())
         x_nan[1, 2, 1] = np.nan
 
         contains_nan, nan_policy_out = _contains_nan(x, nan_policy=nan_policy)