We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5aa1333 commit 691c27bCopy full SHA for 691c27b
array_api_compat/dask/array/_aliases.py
@@ -48,11 +48,7 @@ def astype(
48
copy: bool = True,
49
device: Device | None = None
50
) -> Array:
51
- if device is not None:
52
- raise NotImplementedError(
53
- "The device keyword is not implemented yet for "
54
- "array-api-compat wrapped dask"
55
- )
+ # TODO: respect device keyword?
56
if not copy and dtype == x.dtype:
57
return x
58
# dask astype doesn't respect copy=True,
0 commit comments