From 1ecaba1ebb716faf297c2a328b167d7f28cf9766 Mon Sep 17 00:00:00 2001
From: kapoorlab <varun.kapoor@kapoorlabs.org>
Date: Sat, 11 May 2024 10:27:55 +0000
Subject: [PATCH] pure cell

---
 src/vollseg/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/vollseg/utils.py b/src/vollseg/utils.py
index 303bb45..39e97ce 100644
--- 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(