support endoscopy bundle over windows (#1224)

Signed-off-by: Sachidanand Alle <sachidanand.alle@gmail.com> Signed-off-by: Sachidanand Alle <sachidanand.alle@gmail.com>
Project-MONAI · Dec 20, 2022 · 40fd783 · 40fd783
1 parent e9b9aa1
commit 40fd783
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/sample-apps/endoscopy/lib/configs/inbody.py b/sample-apps/endoscopy/lib/configs/inbody.py
@@ -16,7 +16,7 @@
 import lib.infers
 import lib.trainers
 from lib.scoring.cvat import CVATRandomScoring
-from monai.bundle import download
+from monai.bundle import download, get_bundle_versions
 
 from monailabel.interfaces.config import TaskConfig
 from monailabel.interfaces.tasks.infer_v2 import InferTask
@@ -34,7 +34,11 @@ def init(self, name: str, model_dir: str, conf: Dict[str, str], planner: Any, **
         super().init(name, model_dir, conf, planner, **kwargs)
 
         bundle_name = "endoscopic_inbody_classification"
-        bundle_version = None
+        repo_owner, repo_name, tag_name = "Project-MONAI/model-zoo/hosting_storage_v1".split("/")
+        bundle_version = get_bundle_versions(bundle_name, repo=f"{repo_owner}/{repo_name}", tag=tag_name)[
+            "latest_version"
+        ]
+
         self.bundle_path = os.path.join(self.model_dir, bundle_name)
         if not os.path.exists(self.bundle_path):
             download(name=bundle_name, version=bundle_version, bundle_dir=self.model_dir)

diff --git a/sample-apps/endoscopy/lib/configs/tooltracking.py b/sample-apps/endoscopy/lib/configs/tooltracking.py
@@ -16,7 +16,7 @@
 import lib.infers
 import lib.trainers
 from lib.scoring.cvat import CVATEpistemicScoring
-from monai.bundle import download
+from monai.bundle import download, get_bundle_versions
 
 from monailabel.interfaces.config import TaskConfig
 from monailabel.interfaces.tasks.infer_v2 import InferTask
@@ -34,7 +34,11 @@ def init(self, name: str, model_dir: str, conf: Dict[str, str], planner: Any, **
         super().init(name, model_dir, conf, planner, **kwargs)
 
         bundle_name = "endoscopic_tool_segmentation"
-        bundle_version = None
+        repo_owner, repo_name, tag_name = "Project-MONAI/model-zoo/hosting_storage_v1".split("/")
+        bundle_version = get_bundle_versions(bundle_name, repo=f"{repo_owner}/{repo_name}", tag=tag_name)[
+            "latest_version"
+        ]
+
         self.bundle_path = os.path.join(self.model_dir, bundle_name)
         if not os.path.exists(self.bundle_path):
             download(name=bundle_name, version=bundle_version, bundle_dir=self.model_dir)