Skip to content

Commit 23eb764

Browse files
committed
Rename DASK_DEVICE to _DASK_DEVICE
1 parent 72919ed commit 23eb764

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

array_api_compat/common/_helpers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,7 @@ class _dask_device:
166166
def __repr__(self):
167167
return "DASK_DEVICE"
168168

169-
DASK_DEVICE = _dask_device()
169+
_DASK_DEVICE = _dask_device()
170170

171171
# device() is not on numpy.ndarray or dask.array and to_device() is not on numpy.ndarray
172172
# or cupy.ndarray. They are not included in array objects of this library
@@ -199,7 +199,7 @@ def device(x: Array, /) -> Device:
199199
return "cpu"
200200
except ImportError:
201201
pass
202-
return DASK_DEVICE
202+
return _DASK_DEVICE
203203
elif is_jax_array(x):
204204
# JAX has .device() as a method, but it is being deprecated so that it
205205
# can become a property, in accordance with the standard. In order for

0 commit comments

Comments
 (0)