fix axis typing

Author: FynnBe

src/bioimageio/core/axis.py

@@ -50,7 +50,7 @@ class AxisDescrLike(Protocol):
     type: Literal["batch", "channel", "index", "space", "time"]
 
 
-AxisLike = Union[_AxisLikePlain, AxisDescrLike]
+AxisLike = Union[_AxisLikePlain, AxisDescrLike, v0_5.AnyAxis, "Axis"]
 
 
 @dataclass

src/bioimageio/core/tensor.py

@@ -25,7 +25,7 @@
 from typing_extensions import Self, assert_never
 
 from ._magic_tensor_ops import MagicTensorOpsMixin
-from .axis import Axis, AxisDescrLike, AxisId, AxisInfo, AxisLike, PerAxis
+from .axis import AxisId, AxisInfo, AxisLike, PerAxis
 from .common import (
     CropWhere,
     DTypeStr,
@@ -186,10 +186,10 @@ def from_numpy(
 
         if dims is None:
             return cls._interprete_array_wo_known_axes(array)
-        elif isinstance(dims, (AxisId, AxisDescrLike)):
-            dim_seq = [dims]
-        else:
+        elif isinstance(dims, collections.abc.Sequence):
             dim_seq = list(dims)
+        else:
+            dim_seq = [dims]
 
         axis_infos = [AxisInfo.create(a) for a in dim_seq]
         original_shape = tuple(array.shape)