pyright fix private import errors

Author: FynnBe

bioimageio/core/backends/keras_backend.py

@@ -4,17 +4,18 @@
 from loguru import logger
 from numpy.typing import NDArray
 
-from bioimageio.spec._internal.io import download
-from bioimageio.spec._internal.type_guards import is_list, is_tuple
 from bioimageio.spec.model import v0_4, v0_5
 from bioimageio.spec.model.v0_5 import Version
+from bioimageio.spec.utils import download
 
 from .._settings import settings
 from ..digest_spec import get_axes_infos
+from ..utils._type_guards import is_list, is_tuple
 from ._model_adapter import ModelAdapter
 
 os.environ["KERAS_BACKEND"] = settings.keras_backend
 
+
 # by default, we use the keras integrated with tensorflow
 # TODO: check if we should prefer keras
 try:

bioimageio/core/backends/onnx_backend.py

@@ -5,11 +5,11 @@
 import onnxruntime as rt  # pyright: ignore[reportMissingTypeStubs]
 from numpy.typing import NDArray
 
-from bioimageio.spec._internal.type_guards import is_list, is_tuple
 from bioimageio.spec.model import v0_4, v0_5
 from bioimageio.spec.utils import download
 
 from ..model_adapters import ModelAdapter
+from ..utils._type_guards import is_list, is_tuple
 
 
 class ONNXModelAdapter(ModelAdapter):

bioimageio/core/backends/pytorch_backend.py

@@ -11,13 +11,13 @@
 from torch import nn
 from typing_extensions import assert_never
 
-from bioimageio.spec._internal.type_guards import is_list, is_ndarray, is_tuple
 from bioimageio.spec._internal.version_type import Version
 from bioimageio.spec.common import ZipPath
 from bioimageio.spec.model import AnyModelDescr, v0_4, v0_5
 from bioimageio.spec.utils import download
 
 from ..digest_spec import import_callable
+from ..utils._type_guards import is_list, is_ndarray, is_tuple
 from ._model_adapter import ModelAdapter
 
 

bioimageio/core/backends/torchscript_backend.py

@@ -6,11 +6,11 @@
 import torch
 from numpy.typing import NDArray
 
-from bioimageio.spec._internal.type_guards import is_list, is_tuple
 from bioimageio.spec.model import v0_4, v0_5
 from bioimageio.spec.utils import download
 
 from ..model_adapters import ModelAdapter
+from ..utils._type_guards import is_list, is_tuple
 
 
 class TorchscriptModelAdapter(ModelAdapter):

bioimageio/core/utils/_type_guards.py

@@ -0,0 +1,10 @@
+"""use these type guards with caution!
+They widen the type to T[Any], which is not always correct."""
+
+from bioimageio.spec._internal.type_guards import (
+    is_list,  # pyright: ignore[reportPrivateImportUsage]
+    is_ndarray,  # pyright: ignore[reportPrivateImportUsage]
+    is_tuple,  # pyright: ignore[reportPrivateImportUsage]
+)
+
+__all__ = ["is_list", "is_ndarray", "is_tuple"]