Skip to content

BUG: import cupy_xarray does not work with xarray v2022.11+ #23

@kadykov

Description

@kadykov

It seems that xarray commit broke dask_array_type import from xarray.core.pycompat

import cupy as cp
import cupy_xarray  # Adds .cupy to Xarray objects
import numpy as np
import xarray as xr

gives the following error:

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
Cell In[1], line 5
      3 import xarray as xr
      4 import numpy as np
----> 5 import cupy_xarray  # Adds .cupy to Xarray objects

File c:\ProgramData\Miniconda3\envs\cupy-xarray-original\lib\site-packages\cupy_xarray\__init__.py:2
      [1](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/__init__.py?line=0) from . import _version
----> [2](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/__init__.py?line=1) from .accessors import CupyDataArrayAccessor, CupyDatasetAccessor  # noqa
      [4](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/__init__.py?line=3) __version__ = _version.get_versions()["version"]

File c:\ProgramData\Miniconda3\envs\cupy-xarray-original\lib\site-packages\cupy_xarray\accessors.py:8
      [1](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=0) import cupy as cp
      [2](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=1) from xarray import (
      [3](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=2)     DataArray,
      [4](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=3)     Dataset,
      [5](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=4)     register_dataarray_accessor,
      [6](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=5)     register_dataset_accessor,
      [7](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=6) )
----> [8](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=7) from xarray.core.pycompat import dask_array_type
     [11](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=10) @register_dataarray_accessor("cupy")
     [12](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=11) class CupyDataArrayAccessor:
     [13](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=12)     """
     [14](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=13)     Access methods for DataArrays using Cupy.
     [15](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=14)     Methods and attributes can be accessed through the `.cupy` attribute.
     [16](file:///c%3A/ProgramData/Miniconda3/envs/cupy-xarray-original/lib/site-packages/cupy_xarray/accessors.py?line=15)     """

ImportError: cannot import name 'dask_array_type' from 'xarray.core.pycompat' (c:\ProgramData\Miniconda3\envs\cupy-xarray-original\lib\site-packages\xarray\core\pycompat.py)

python version: 3.9.15
cupy version: 11.4.0
xarray version: 2202.12.0
dask version: 2022.12.1
cupy-xarray version: 0.1.2

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions