fix: correctly handle NaNs when doing auto cca

SchmollerLab · Jun 15, 2023 · 0808a3c · 0808a3c
1 parent 08eb8a9
commit 0808a3c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cellacdc/gui.py b/cellacdc/gui.py
@@ -15019,7 +15019,10 @@ def autoCca_df(self, enforceAll=False):
         # For the last visited frame we perform assignment again only on
         # IDs where we didn't manually correct assignment
         if lastVisited and not enforceAll:
-            correctedAssignIDs = curr_df[curr_df['corrected_assignment']].index
+            try:
+                correctedAssignIDs = curr_df[curr_df['corrected_assignment']].index
+            except Exception as e:
+                correctedAssignIDs = []
             posData.new_IDs = [
                 ID for ID in posData.new_IDs
                 if ID not in correctedAssignIDs