unload data on iteration over leaves (#1534)

sourmash-bio · May 18, 2021 · a782b1d · a782b1d
1 parent 5234bf1
commit a782b1d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/sourmash/sbt.py b/src/sourmash/sbt.py
@@ -1120,12 +1120,14 @@ def _parents(self, pos=0):
                 yield p.pos
                 p = self.parent(p.pos)
 
-    def leaves(self, with_pos=False):
+    def leaves(self, with_pos=False, unload_data=True):
         for pos, data in self._leaves.items():
             if with_pos:
                 yield (pos, data)
             else:
                 yield data
+            if unload_data:
+                data.unload()
 
     def combine(self, other):
         larger, smaller = self, other