ENH: New functions lazy_raise, lazy_warn, and lazy_wait_on

Author: crusaderky

docs/api-lazy.md

@@ -10,6 +10,9 @@ lazy backends, e.g. Dask or Jax:
     :toctree: generated
 
     lazy_apply
+    lazy_raise
+    lazy_wait_on
+    lazy_warn
     testing.lazy_xp_function
     testing.patch_lazy_xp_functions
 ```

docs/conf.py

@@ -56,6 +56,7 @@
     "numpy": ("https://numpy.org/doc/stable", None),
     "dask": ("https://docs.dask.org/en/stable", None),
     "jax": ("https://jax.readthedocs.io/en/latest", None),
+    "equinox": ("https://docs.kidger.site/equinox/", None),
 }
 
 nitpick_ignore = [

pixi.lock

@@ -26,7 +26,7 @@ classifiers = [
     "Typing :: Typed",
 ]
 dynamic = ["version"]
-dependencies = ["array-api-compat>=1.10.0,<2"]
+# dependencies = ["array-api-compat>=1.10.0,<2"]  # DNM
 
 [project.urls]
 Homepage = "https://github.com/data-apis/array-api-extra"
@@ -48,10 +48,11 @@ platforms = ["linux-64", "osx-arm64", "win-64"]
 
 [tool.pixi.dependencies]
 python = ">=3.10,<3.14"
-array-api-compat = ">=1.10.0,<2"
+# array-api-compat = ">=1.10.0,<2"  # DNM
 
 [tool.pixi.pypi-dependencies]
 array-api-extra = { path = ".", editable = true }
+array-api-compat = { git = "https://github.com/data-apis/array-api-compat" }  # DNM
 
 [tool.pixi.feature.lint.dependencies]
 typing-extensions = "*"

pyproject.toml

@@ -12,7 +12,7 @@
     setdiff1d,
     sinc,
 )
-from ._lib._lazy import lazy_apply
+from ._lib._lazy import lazy_apply, lazy_raise, lazy_wait_on, lazy_warn
 
 __version__ = "0.6.1.dev0"
 
@@ -27,6 +27,9 @@
     "isclose",
     "kron",
     "lazy_apply",
+    "lazy_raise",
+    "lazy_wait_on",
+    "lazy_warn",
     "nunique",
     "pad",
     "setdiff1d",