remove useless DaskArray type in scalar or array type

pydata · Jul 12, 2024 · 27db395 · 27db395
1 parent ab4dbf5
commit 27db395
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/xarray/core/types.py b/xarray/core/types.py
@@ -178,7 +178,7 @@ def copy(
 T_ExtensionArray = TypeVar("T_ExtensionArray", bound=pd.api.extensions.ExtensionArray)
 
 
-ScalarOrArray = Union["ArrayLike", np.generic, np.ndarray, "DaskArray"]
+ScalarOrArray = Union["ArrayLike", np.generic]
 VarCompatible = Union["Variable", "ScalarOrArray"]
 DaCompatible = Union["DataArray", "VarCompatible"]
 DsCompatible = Union["Dataset", "DaCompatible"]

diff --git a/xarray/tests/test_groupby.py b/xarray/tests/test_groupby.py
@@ -1481,7 +1481,7 @@ def test_groupby_math(self) -> None:
         with pytest.raises(TypeError, match=r"only support binary ops"):
             grouped + grouped  # type: ignore[type-var]
         with pytest.raises(TypeError, match=r"in-place operations"):
-            array += grouped
+            array += grouped  # type: ignore[arg-type]
 
     def test_groupby_math_not_aligned(self) -> None:
         array = DataArray(