(chore): more robust size getting

scverse · Jan 27, 2025 · 71c680d · 71c680d
1 parent cce570f
commit 71c680d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/anndata/_core/anndata.py b/src/anndata/_core/anndata.py
@@ -97,10 +97,10 @@ def _infer_shape_for_axis(
     for elem in [layers, xxxm, xxxp]:
         if elem is not None:
             elem = cast(Mapping, elem)
-            next_elem = next(iter(elem.values()))
-            if hasattr(next_elem, "shape"):
-                size = cast(int, next_elem.shape[0])
-                return size
+            for sub_elem in elem.values():
+                if hasattr(sub_elem, "shape"):
+                    size = cast(int, sub_elem.shape[0])
+                    return size
     return None