Skip to content

Commit

Permalink
Merge branch 'dev' into add-maisi
Browse files Browse the repository at this point in the history
  • Loading branch information
yiheng-wang-nv committed Aug 2, 2024
2 parents 69dd9ff + f477960 commit 4108dc9
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 0 deletions.
24 changes: 24 additions & 0 deletions ci/unit_tests/test_vista3d.py
Original file line number Diff line number Diff line change
Expand Up @@ -275,8 +275,12 @@ def test_train_config(self, override):
override["val_datalist"] = val_datalist

bundle_root = override["bundle_root"]
<<<<<<< HEAD
if bundle_root not in sys.path:
sys.path = [bundle_root] + sys.path
=======
sys.path = [bundle_root] + sys.path
>>>>>>> dev
trainer = ConfigWorkflow(
workflow_type="train",
config_file=os.path.join(bundle_root, "configs/train.json"),
Expand Down Expand Up @@ -306,8 +310,12 @@ def test_eval_config(self, override):
override["train_datalist"] = train_datalist
override["val_datalist"] = val_datalist
bundle_root = override["bundle_root"]
<<<<<<< HEAD
if bundle_root not in sys.path:
sys.path = [bundle_root] + sys.path
=======
sys.path = [bundle_root] + sys.path
>>>>>>> dev
config_files = [
os.path.join(bundle_root, "configs/train.json"),
os.path.join(bundle_root, "configs/train_continual.json"),
Expand Down Expand Up @@ -344,8 +352,12 @@ def test_train_continual_config(self, override):
override["val_datalist"] = val_datalist

bundle_root = override["bundle_root"]
<<<<<<< HEAD
if bundle_root not in sys.path:
sys.path = [bundle_root] + sys.path
=======
sys.path = [bundle_root] + sys.path
>>>>>>> dev
trainer = ConfigWorkflow(
workflow_type="train",
config_file=[
Expand Down Expand Up @@ -376,8 +388,12 @@ def test_infer_config(self, override):
override["input_dict"] = input_dict

bundle_root = override["bundle_root"]
<<<<<<< HEAD
if bundle_root not in sys.path:
sys.path = [bundle_root] + sys.path
=======
sys.path = [bundle_root] + sys.path
>>>>>>> dev

inferrer = ConfigWorkflow(
workflow_type="infer",
Expand All @@ -400,8 +416,12 @@ def test_batch_infer_config(self, override):
params["input_suffix"] = "image_*.nii.gz"

bundle_root = override["bundle_root"]
<<<<<<< HEAD
if bundle_root not in sys.path:
sys.path = [bundle_root] + sys.path
=======
sys.path = [bundle_root] + sys.path
>>>>>>> dev
config_files = [
os.path.join(bundle_root, "configs/inference.json"),
os.path.join(bundle_root, "configs/batch_inference.json"),
Expand All @@ -424,8 +444,12 @@ def test_error_prompt_infer_config(self, override):
override["input_dict"] = input_dict

bundle_root = override["bundle_root"]
<<<<<<< HEAD
if bundle_root not in sys.path:
sys.path = [bundle_root] + sys.path
=======
sys.path = [bundle_root] + sys.path
>>>>>>> dev

inferrer = ConfigWorkflow(
workflow_type="infer",
Expand Down
4 changes: 4 additions & 0 deletions models/model_info.json
Original file line number Diff line number Diff line change
Expand Up @@ -1522,5 +1522,9 @@
"spleen_ct_segmentation_v0.5.8": {
"checksum": "3e43f860ba08cb3d121d521cadde0e5b59f72526",
"source": "https://api.ngc.nvidia.com/v2/models/nvidia/monaihosting/spleen_ct_segmentation/versions/0.5.8/files/spleen_ct_segmentation_v0.5.8.zip"
},
"vista3d_v0.4.1": {
"checksum": "499d6b79d351bdd611ded0a7780d7803096e5f62",
"source": "https://api.ngc.nvidia.com/v2/models/nvidia/monaihosting/vista3d/versions/0.4.1/files/vista3d_v0.4.1.zip"
}
}

0 comments on commit 4108dc9

Please sign in to comment.