update single_cell.py: fix a bug when mask dataset is present

xing-lab-pitt · May 1, 2024 · 971ed3a · 971ed3a
1 parent 2902bd4
commit 971ed3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/livecellx/core/single_cell.py b/livecellx/core/single_cell.py
@@ -256,7 +256,7 @@ def get_img(self):
         return self.img_dataset.get_img_by_time(self.timeframe)
 
     def get_mask(self, dtype=bool):
-        if isinstance(self.mask_dataset, SingleImageDataset):
+        if isinstance(self.mask_dataset, SingleImageDataset) or isinstance(self.mask_dataset, LiveCellImageDataset):
             try:
                 return self.mask_dataset.get_img_by_time()
             except Exception as e: