Skip to content

Commit b718c90

Browse files
committed
fix types
1 parent 62b1b26 commit b718c90

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

xarray/coding/variables.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
)
2222
from xarray.coding.times import CFDatetimeCoder, CFTimedeltaCoder
2323
from xarray.core import dtypes, duck_array_ops, indexing
24+
from xarray.core.types import Self
2425
from xarray.core.variable import Variable
2526

2627
if TYPE_CHECKING:
@@ -63,7 +64,7 @@ def _oindex_get(self, key):
6364
def _vindex_get(self, key):
6465
return type(self)(self.array.vindex[key])
6566

66-
def __getitem__(self, key) -> np.ndarray:
67+
def __getitem__(self, key) -> Self:
6768
return type(self)(self.array[key])
6869

6970
def get_duck_array(self):
@@ -104,7 +105,7 @@ def _oindex_get(self, key):
104105
def _vindex_get(self, key):
105106
return type(self)(self.array.vindex[key])
106107

107-
def __getitem__(self, key) -> np.ndarray:
108+
def __getitem__(self, key) -> Self:
108109
return type(self)(self.array[key])
109110

110111
def get_duck_array(self):

xarray/core/indexes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -717,7 +717,7 @@ def from_variables(
717717

718718
# preserve wrapped pd.Index (if any)
719719
# accessing `.data` can load data from disk, so we only access if needed
720-
data = var._data if isinstance(var._data, PandasIndexingAdapter) else var.data
720+
data = var._data if isinstance(var._data, PandasIndexingAdapter) else var.data # type: ignore[redundant-expr]
721721
# multi-index level variable: get level index
722722
if isinstance(var._data, PandasMultiIndexingAdapter):
723723
level = var._data.level

xarray/tests/test_dtypes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ class DummyArrayAPINamespace:
1515
int32 = None # type: ignore[unused-ignore,var-annotated]
1616
float64 = None # type: ignore[unused-ignore,var-annotated]
1717

18-
array_api_strict = DummyArrayAPINamespace
18+
array_api_strict = DummyArrayAPINamespace # type: ignore[assignment]
1919

2020

2121
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)