allow space and time axes to be singletons

bioimage-io · Aug 21, 2024 · e5d83ba · e5d83ba
1 parent f6c3bd7
commit e5d83ba
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/bioimageio/core/axis.py b/bioimageio/core/axis.py
@@ -80,10 +80,8 @@ def create(cls, axis: AxisLike, maybe_singleton: Optional[bool] = None) -> AxisI
 
         axis_base = super().create(axis)
         if maybe_singleton is None:
-            if isinstance(axis, Axis):
-                maybe_singleton = axis.type in ("batch", "channel", "index")
-            elif isinstance(axis, str):
-                maybe_singleton = axis in ("b", "c", "i")
+            if isinstance(axis, (Axis, str)):
+                maybe_singleton = True
             else:
                 if axis.size is None:
                     maybe_singleton = True