fall back to adj.leaves if particles.final missing #174

jacanchaplais · Mar 1, 2024 · 6402157 · 6402157
1 parent 6428304
commit 6402157
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/graphicle/data.py b/graphicle/data.py
@@ -2751,7 +2751,10 @@ def hard_mask(self) -> MaskGroup:
     @property
     def final(self) -> MaskArray:
         """Boolean array indicating final state in particle set."""
-        return self.particles.final
+        data = self.particles.final
+        if (not data) and self.adj:
+            return self.adj.leaves
+        return data
 
     @property
     def edges(self) -> base.VoidVector: