small fix and rollback array-api-compat hack

Author: lithomas1

scipy/_lib/_util.py

@@ -12,7 +12,7 @@
 import numpy as np
 from scipy._lib._array_api import array_namespace, is_numpy, xp_size
 from scipy._lib._docscrape import FunctionDoc, Parameter
-from scipy._lib.array_api import is_dask, is_jax
+from scipy._lib._array_api import is_dask, is_jax
 
 AxisError: type[Exception]
 ComplexWarning: type[Warning]
@@ -127,7 +127,7 @@ def _lazywhere(cond, arrays, f, fillvalue=None, f2=None):
     """
     xp = array_namespace(cond, *arrays)
 
-    if is_dask_namespace(xp) or is_jax_namespace(xp):
+    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)