AxisId is also AxisLike!

Author: FynnBe

Diff for: bioimageio/core/axis.py

@@ -34,7 +34,7 @@ def _get_axis_type(a: Literal["b", "t", "i", "c", "x", "y", "z"]):
 BatchSize = int
 
 AxisLetter = Literal["b", "i", "t", "c", "z", "y", "x"]
-AxisLike = Union[AxisLetter, v0_5.AnyAxis, "Axis"]
+AxisLike = Union[AxisId, AxisLetter, v0_5.AnyAxis, "Axis"]
 
 
 @dataclass