Skip to content

Commit 5d367db

Browse files
committed
force object dtype in asarray
1 parent 2a3965c commit 5d367db

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

xarray/core/dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1485,7 +1485,7 @@ def __getitem__(self, key):
14851485
if hashable(key):
14861486
return self._construct_dataarray(key)
14871487
else:
1488-
return self._copy_listed(np.asarray(key))
1488+
return self._copy_listed(np.asarray(key, dtype="O"))
14891489

14901490
def __setitem__(self, key: Union[Hashable, List[Hashable], Mapping], value) -> None:
14911491
"""Add an array to this dataset.

xarray/tests/test_dataset.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3305,6 +3305,11 @@ def test_getitem_hashable(self):
33053305
with pytest.raises(KeyError, match=r"('var1', 'var2')"):
33063306
data[("var1", "var2")]
33073307

3308+
def test_getitem_multiple_dtype(self):
3309+
keys = ["foo", 1]
3310+
dataset = Dataset({key: ("dim0", range(1)) for key in keys})
3311+
assert_identical(dataset, dataset[keys])
3312+
33083313
def test_virtual_variables_default_coords(self):
33093314
dataset = Dataset({"foo": ("x", range(10))})
33103315
expected = DataArray(range(10), dims="x", name="x")

0 commit comments

Comments
 (0)