more elegant fix

scverse · Dec 19, 2024 · b4a4905 · b4a4905
1 parent 4b65193
commit b4a4905
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/anndata/compat/__init__.py b/src/anndata/compat/__init__.py
@@ -47,7 +47,7 @@ class Empty:
     pass
 
 
-Index1D = slice | int | str | np.int64 | np.ndarray
+Index1D = slice | int | str | np.int64 | np.ndarray | pd.Series[bool]
 IndexRest = Index1D | EllipsisType
 Index = (
     IndexRest

diff --git a/tests/test_concatenate.py b/tests/test_concatenate.py
@@ -1444,7 +1444,7 @@ def test_concat_outer_aligned_mapping(elem):
     del b.obsm[elem]
 
     concated = concat({"a": a, "b": b}, join="outer", label="group")
-    result = concated.obsm[elem][(concated.obs["group"] == "b").to_numpy()]
+    result = concated[concated.obs["group"] == "b"].obsm[elem]
 
     check_filled_like(result, elem_name=f"obsm/{elem}")