8000 Add overload for Mapping behavior · nbren12/xarray@414ee30 · GitHub
[go: up one dir, main page]

Skip to content

Commit 414ee30

Browse files
committed
Add overload for Mapping behavior
Sadly this is not working with my version of mypy. See python/mypy#7328
1 parent 34e3ce0 commit 414ee30
8000

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

xarray/core/dataset.py

Lines changed: 8 additions & 3 deletions
93B7
Original file line numberDiff line numberDiff line change
@@ -1241,13 +1241,18 @@ def loc(self) -> _LocIndexer:
12411241
and only when the key is a dict of the form {dim: labels}.
12421242
"""
12431243
return _LocIndexer(self)
1244-
12451244

12461245
@overload
1247-
def __getitem__(self, key: Hashable) -> DataArray: ...
1246+
def __getitem__(self, key: Hashable) -> DataArray:
1247+
...
12481248

12491249
@overload
1250-
def __getitem__(self, key: Any) -> Dataset: ...
1250+
def __getitem__(self, key: Mapping) -> "Dataset":
1251+
...
1252+
1253+
@overload
1254+
def __getitem__(self, key: List) -> "Dataset":
1255+
...
12511256

12521257
def __getitem__(self, key):
12531258
"""Access variables or coordinates this dataset as a

0 commit comments

Comments
 (0)
0