From 23eb7649cb5ac08cb231256c7a2491af434769f7 Mon Sep 17 00:00:00 2001 From: Aaron Meurer Date: Mon, 26 Feb 2024 15:44:52 -0700 Subject: [PATCH] Rename DASK_DEVICE to _DASK_DEVICE --- array_api_compat/common/_helpers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/array_api_compat/common/_helpers.py b/array_api_compat/common/_helpers.py index 5cb9800f..bd2372b5 100644 --- a/array_api_compat/common/_helpers.py +++ b/array_api_compat/common/_helpers.py @@ -166,7 +166,7 @@ class _dask_device: def __repr__(self): return "DASK_DEVICE" -DASK_DEVICE = _dask_device() +_DASK_DEVICE = _dask_device() # device() is not on numpy.ndarray or dask.array and to_device() is not on numpy.ndarray # or cupy.ndarray. They are not included in array objects of this library @@ -199,7 +199,7 @@ def device(x: Array, /) -> Device: return "cpu" except ImportError: pass - return DASK_DEVICE + return _DASK_DEVICE elif is_jax_array(x): # JAX has .device() as a method, but it is being deprecated so that it # can become a property, in accordance with the standard. In order for