Skip to content

Commit ad41e7a

Browse files
authored
Update test for CuPy v9 (#5369)
1 parent de6144c commit ad41e7a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

xarray/tests/test_cupy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def test_cupy_import():
4747
def test_check_data_stays_on_gpu(toy_weather_data):
4848
"""Perform some operations and check the data stays on the GPU."""
4949
freeze = (toy_weather_data["tmin"] <= 0).groupby("time.month").mean("time")
50-
assert isinstance(freeze.data, cp.core.core.ndarray)
50+
assert isinstance(freeze.data, cp.ndarray)
5151

5252

5353
def test_where():

0 commit comments

Comments
 (0)