Merge pull request #559 from bioimage-io/local_n

minor improvements

bioimageio/spec/common.py

@@ -7,9 +7,6 @@
 from ._internal.io import Sha256 as Sha256
 from ._internal.io import YamlValue as YamlValue
 from ._internal.io_basics import FileName as FileName
-from ._internal.io_utils import (
-    identify_bioimageio_yaml_file_name as identify_bioimageio_yaml_file_name,
-)
 from ._internal.root_url import RootHttpUrl as RootHttpUrl
 from ._internal.types import FileSource as FileSource
 from ._internal.types import PermissiveFileSource as PermissiveFileSource

bioimageio/spec/model/v0_5.py

@@ -221,12 +221,10 @@ def _validate(cls, value: str):
 
 
 class ParameterizedSize(Node):
-    """Describes a range of valid tensor axis sizes as `size = min + n*step`.
-    `n` in this equation is the same for all axis parameterized in this manner across the whole model.
-    """
+    """Describes a range of valid tensor axis sizes as `size = min + n*step`."""
 
     N: ClassVar[Type[int]] = int
-    """integer to parameterize all axes with a `ParameterizedSize`"""
+    """integer to parameterize this axis"""
 
     min: Annotated[int, Gt(0)]
     step: Annotated[int, Gt(0)]

bioimageio/spec/utils.py

@@ -1,3 +1,6 @@
 from ._internal.io import download as download
+from ._internal.io_utils import (
+    identify_bioimageio_yaml_file_name as identify_bioimageio_yaml_file_name,
+)
 from ._internal.io_utils import load_array as load_array
 from ._internal.io_utils import save_array as save_array