pyright fixes

Author: FynnBe

bioimageio/core/backends/keras_backend.py

@@ -27,6 +27,8 @@
 except Exception:
     import keras
 
+    tf_version = None
+
 
 class KerasModelAdapter(ModelAdapter):
     def __init__(

bioimageio/core/backends/onnx_backend.py

@@ -1,3 +1,4 @@
+# pyright: reportUnknownVariableType=false
 import warnings
 from typing import Any, List, Optional, Sequence, Union
 
@@ -25,8 +26,8 @@ def __init__(
 
         local_path = download(model_description.weights.onnx.source).path
         self._session = rt.InferenceSession(local_path.read_bytes())
-        onnx_inputs = self._session.get_inputs()  # type: ignore
-        self._input_names: List[str] = [ipt.name for ipt in onnx_inputs]  # type: ignore
+        onnx_inputs = self._session.get_inputs()
+        self._input_names: List[str] = [ipt.name for ipt in onnx_inputs]
 
         if devices is not None:
             warnings.warn(
@@ -40,11 +41,11 @@ def _forward_impl(
             None, dict(zip(self._input_names, input_arrays))
         )
         if is_list(result) or is_tuple(result):
-            result_seq = result
+            result_seq = list(result)
         else:
             result_seq = [result]
 
-        return result_seq  # pyright: ignore[reportReturnType]
+        return result_seq
 
     def unload(self) -> None:
         warnings.warn(

bioimageio/core/backends/pytorch_backend.py

@@ -146,10 +146,16 @@ def load_torch_state_dict(
         state = torch.load(f, map_location=devices[0], weights_only=True)
 
     incompatible = model.load_state_dict(state)
-    if incompatible is not None and incompatible.missing_keys:
+    if (
+        incompatible is not None  # pyright: ignore[reportUnnecessaryComparison]
+        and incompatible.missing_keys
+    ):
         logger.warning("Missing state dict keys: {}", incompatible.missing_keys)
 
-    if incompatible is not None and incompatible.unexpected_keys:
+    if (
+        incompatible is not None  # pyright: ignore[reportUnnecessaryComparison]
+        and incompatible.unexpected_keys
+    ):
         logger.warning("Unexpected state dict keys: {}", incompatible.unexpected_keys)
 
     return model

bioimageio/core/backends/torchscript_backend.py

@@ -1,3 +1,4 @@
+# pyright: reportUnknownVariableType=false
 import gc
 import warnings
 from typing import Any, List, Optional, Sequence, Union