is_lazy_array

Author: crusaderky

src/array_api_extra/_lib/_utils/_compat.py

@@ -14,6 +14,7 @@
         is_dask_namespace,
         is_jax_array,
         is_jax_namespace,
+        is_lazy_array,
         is_numpy_array,
         is_numpy_namespace,
         is_pydata_sparse_array,
@@ -35,6 +36,7 @@
         is_dask_namespace,
         is_jax_array,
         is_jax_namespace,
+        is_lazy_array,
         is_numpy_array,
         is_numpy_namespace,
         is_pydata_sparse_array,
@@ -56,6 +58,7 @@
     "is_dask_namespace",
     "is_jax_array",
     "is_jax_namespace",
+    "is_lazy_array",
     "is_numpy_array",
     "is_numpy_namespace",
     "is_pydata_sparse_array",

src/array_api_extra/_lib/_utils/_compat.pyi

@@ -32,5 +32,6 @@ def is_jax_array(x: object, /) -> bool: ...
 def is_numpy_array(x: object, /) -> bool: ...
 def is_pydata_sparse_array(x: object, /) -> bool: ...
 def is_torch_array(x: object, /) -> bool: ...
+def is_lazy_array(x: object, /) -> bool: ...
 def is_writeable_array(x: object, /) -> bool: ...
 def size(x: Array, /) -> int | None: ...

vendor_tests/test_vendor.py

@@ -14,6 +14,7 @@ def test_vendor_compat():
         is_dask_namespace,
         is_jax_array,
         is_jax_namespace,
+        is_lazy_array,
         is_numpy_array,
         is_numpy_namespace,
         is_pydata_sparse_array,
@@ -35,6 +36,7 @@ def test_vendor_compat():
     assert not is_dask_namespace(xp)
     assert not is_jax_array(x)
     assert not is_jax_namespace(xp)
+    assert not is_lazy_array(x)
     assert not is_numpy_array(x)
     assert not is_numpy_namespace(xp)
     assert not is_pydata_sparse_array(x)