add dependencies field to tensorflow_saved_model_bundle weights

Author: FynnBe

bioimageio/spec/_internal/validation_context.py

@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Any, Dict, Literal, Optional, Tuple, Union
+from typing import Any, Dict, Literal, Optional, Union
 
 from pydantic import AnyUrl, BaseModel, DirectoryPath
 from typing_extensions import NotRequired, TypedDict

bioimageio/spec/model/v0_5.py

@@ -880,13 +880,14 @@ class PytorchStateDictWeights(WeightsEntryBase, frozen=True):
     type = "pytorch_state_dict"
     weights_format_name: ClassVar[str] = "Pytorch State Dict"
     architecture: Architecture
-    dependencies: Optional[Dependencies] = None
-    """Custom dependencies beyond pytorch.
-    Should include pytorch and any version pinning has to be compatible with `pytorch_version`."""
     pytorch_version: Version
     """Version of the PyTorch library used.
     If `depencencies` is specified it has to include pytorch and any version pinning has to be compatible."""
 
+    dependencies: Optional[Dependencies] = None
+    """Custom dependencies beyond pytorch.
+    Should include pytorch and any version pinning has to be compatible with `pytorch_version`."""
+
 
 class TensorflowJsWeights(WeightsEntryBase, frozen=True):
     type = "tensorflow_js"
@@ -905,6 +906,10 @@ class TensorflowSavedModelBundleWeights(WeightsEntryBase, frozen=True):
     tensorflow_version: Version
     """Version of the TensorFlow library used."""
 
+    dependencies: Optional[Dependencies] = None
+    """Custom dependencies beyond tensorflow.
+    Should include tensorflow and any version pinning has to be compatible with `tensorflow_version`."""
+
     source: Union[HttpUrl, RelativeFilePath]
     """∈📦 The multi-file weights.
     All required files/folders should be a zip archive."""