pure cell

Kapoorlabs-CAPED · May 11, 2024 · 1ecaba1 · 1ecaba1
1 parent 7f9ed01
commit 1ecaba1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/vollseg/utils.py b/src/vollseg/utils.py
@@ -1526,7 +1526,7 @@ def CellPoseSeg(
     if save_dir is not None:
         Path(save_dir).mkdir(exist_ok=True)
 
-        if cellpose_model_path is not None:
+        if cellpose_model_path is not None  or cellpose_model_type is not None:
             cellpose_results = Path(save_dir) / "CellPose"
             Path(cellpose_results).mkdir(exist_ok=True)
             imwrite(