|
12 | 12 |
|
13 | 13 | import numpy as np |
14 | 14 | import pandas as pd |
15 | | -from numpy.typing import ArrayLike |
16 | 15 | from packaging.version import Version |
17 | 16 |
|
18 | 17 | import xarray as xr # only for Dataset and DataArray |
@@ -244,7 +243,7 @@ def _possibly_convert_objects(values): |
244 | 243 |
|
245 | 244 |
|
246 | 245 | def as_compatible_data( |
247 | | - data: T_DuckArray | ArrayLike, fastpath: bool = False |
| 246 | + data: T_DuckArray | np.typing.ArrayLike, fastpath: bool = False |
248 | 247 | ) -> T_DuckArray: |
249 | 248 | """Prepare and wrap data to put in a Variable. |
250 | 249 |
|
@@ -371,7 +370,7 @@ class Variable(NamedArray, AbstractArray, VariableArithmetic): |
371 | 370 | def __init__( |
372 | 371 | self, |
373 | 372 | dims, |
374 | | - data: T_DuckArray | ArrayLike, |
| 373 | + data: T_DuckArray | np.typing.ArrayLike, |
375 | 374 | attrs=None, |
376 | 375 | encoding=None, |
377 | 376 | fastpath=False, |
@@ -466,7 +465,7 @@ def data(self): |
466 | 465 | return duck_array |
467 | 466 |
|
468 | 467 | @data.setter # type: ignore[override,unused-ignore] |
469 | | - def data(self, data: T_DuckArray | ArrayLike) -> None: |
| 468 | + def data(self, data: T_DuckArray | np.typing.ArrayLike) -> None: |
470 | 469 | data = as_compatible_data(data) |
471 | 470 | self._check_shape(data) |
472 | 471 | self._data = data |
@@ -931,7 +930,7 @@ def drop_encoding(self) -> Self: |
931 | 930 | def _copy( |
932 | 931 | self, |
933 | 932 | deep: bool = True, |
934 | | - data: T_DuckArray | ArrayLike | None = None, |
| 933 | + data: T_DuckArray | np.typing.ArrayLike | None = None, |
935 | 934 | memo: dict[int, Any] | None = None, |
936 | 935 | ) -> Self: |
937 | 936 | if data is None: |
@@ -1914,7 +1913,7 @@ def no_conflicts(self, other, equiv=duck_array_ops.array_notnull_equiv): |
1914 | 1913 |
|
1915 | 1914 | def quantile( |
1916 | 1915 | self, |
1917 | | - q: ArrayLike, |
| 1916 | + q: np.typing.ArrayLike, |
1918 | 1917 | dim: str | Sequence[Hashable] | None = None, |
1919 | 1918 | method: QuantileMethods = "linear", |
1920 | 1919 | keep_attrs: bool | None = None, |
@@ -2872,7 +2871,9 @@ def concat( |
2872 | 2871 |
|
2873 | 2872 | return cls(first_var.dims, data, attrs) |
2874 | 2873 |
|
2875 | | - def copy(self, deep: bool = True, data: T_DuckArray | ArrayLike | None = None): |
| 2874 | + def copy( |
| 2875 | + self, deep: bool = True, data: T_DuckArray | np.typing.ArrayLike | None = None |
| 2876 | + ): |
2876 | 2877 | """Returns a copy of this object. |
2877 | 2878 |
|
2878 | 2879 | `deep` is ignored since data is stored in the form of |
|
0 commit comments