Corrects unittest and verify_bundle to include brain synthesis only

Project-MONAI · Jan 20, 2025 · 9161f0b · 9161f0b
1 parent eb6bbcb
commit 9161f0b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 50 deletions.
diff --git a/ci/unit_tests/test_cxr_image_synthesis_latent_diffusion_model.py b/ci/unit_tests/test_cxr_image_synthesis_latent_diffusion_model.py
diff --git a/ci/verify_bundle.py b/ci/verify_bundle.py
@@ -54,8 +54,8 @@ def _get_weights_names(bundle: str):
     if bundle == "pediatric_abdominal_ct_segmentation":
         # skip test for this bundle's ts file
         return "dynunet_FT.pt", None
-    if bundle == "cxr_image_synthesis_latent_diffusion_model":
-        return "autoencoder.pt", None
+    if bundle == "brain_image_synthesis_latent_diffusion":
+        return "autoencoder.pt", "model.pt"
     return "model.pt", "model.ts"