fix: removed labels will not always have an assigned fill count

seung-lab · Jun 22, 2024 · 37d37cc · 37d37cc
1 parent 037d594
commit 37d37cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastmorph/__init__.py b/fastmorph/__init__.py
@@ -271,7 +271,7 @@ def fill_holes(
 
   if return_fill_count:
     for label in removed_set:
-      del fill_counts[label]
+      fill_counts.pop(label, None)
     ret.append(fill_counts)
 
   if return_removed: