small fix and rollback array-api-compat hack

Author: lithomas1

scipy/_lib/_util.py

@@ -128,7 +128,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)