Skip to content

Commit

Permalink
Add a TODO message for the future to check BMZ model conda environmen…
Browse files Browse the repository at this point in the history
…t.yaml file with its dependencies
  • Loading branch information
danifranco committed Sep 16, 2024
1 parent 82f0663 commit b79a208
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions biapy/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -510,6 +510,10 @@ def check_bmz_model_compatibility(
# Accepting models that are exported in pytorch_state_dict and with just one input
if "pytorch_state_dict" in model_rdf["weights"] and len(model_rdf["inputs"]) == 1:

# TODO: control model.weights.pytorch_state_dict.dependencies conda env to check if all
# dependencies are installed
# https://github.com/bioimage-io/collection-bioimage-io/issues/609

model_version = Version("0.5")
if "format_version" in model_rdf:
model_version = Version(model_rdf["format_version"])
Expand Down

0 comments on commit b79a208

Please sign in to comment.