ignore device

data-apis · Jan 15, 2025 · 691c27b · 691c27b
1 parent 5aa1333
commit 691c27b
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/array_api_compat/dask/array/_aliases.py b/array_api_compat/dask/array/_aliases.py
@@ -48,11 +48,7 @@ def astype(
     copy: bool = True,
     device: Device | None = None
 ) -> Array:
-    if device is not None:
-        raise NotImplementedError(
-            "The device keyword is not implemented yet for "
-            "array-api-compat wrapped dask"
-        )
+    # TODO: respect device keyword?
     if not copy and dtype == x.dtype:
         return x
     # dask astype doesn't respect copy=True,