We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 72919ed commit 23eb764Copy full SHA for 23eb764
array_api_compat/common/_helpers.py
@@ -166,7 +166,7 @@ class _dask_device:
166
def __repr__(self):
167
return "DASK_DEVICE"
168
169
-DASK_DEVICE = _dask_device()
+_DASK_DEVICE = _dask_device()
170
171
# device() is not on numpy.ndarray or dask.array and to_device() is not on numpy.ndarray
172
# or cupy.ndarray. They are not included in array objects of this library
@@ -199,7 +199,7 @@ def device(x: Array, /) -> Device:
199
return "cpu"
200
except ImportError:
201
pass
202
- return DASK_DEVICE
+ return _DASK_DEVICE
203
elif is_jax_array(x):
204
# JAX has .device() as a method, but it is being deprecated so that it
205
# can become a property, in accordance with the standard. In order for
0 commit comments