Skip to content

Commit c34feeb

Browse files
jacobtomlinsonweiji14keewis
authored
Check if Dask is installed (#62)
Fix bug so that an isinstance check doesn't require dask to be installed to work. --------- Co-authored-by: Wei Ji <[email protected]> Co-authored-by: Justus Magin <[email protected]>
1 parent ee95c00 commit c34feeb

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

cupy_xarray/accessors.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from typing import TYPE_CHECKING, Any
2+
13
import cupy as cp
24
from xarray import (
35
DataArray,
@@ -6,12 +8,16 @@
68
register_dataset_accessor,
79
)
810

11+
if TYPE_CHECKING:
12+
DuckArrayTypes = tuple[type[Any], ...]
13+
dask_array_type: DuckArrayTypes
14+
915
try:
1016
import dask.array
1117

12-
dask_array_type = dask.array.Array
18+
dask_array_type = (dask.array.Array,)
1319
except ImportError:
14-
dask_array_type = None
20+
dask_array_type = ()
1521

1622

1723
@register_dataarray_accessor("cupy")

0 commit comments

Comments
 (0)