Skip to content

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

Closed
@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

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions