make 'find' a generator

sourmash-bio · Mar 9, 2021 · 70168f1 · 70168f1
1 parent 6b4be47
commit 70168f1
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/sourmash/index.py b/src/sourmash/index.py
@@ -33,10 +33,9 @@ def find(self, search_fn, *args, **kwargs):
 
         matches = []
 
-        for node in self.signatures():
-            if search_fn(node, *args):
-                matches.append(node)
-        return matches
+        for ss in self.signatures():
+            if search_fn(ss, *args):
+                yield ss
 
     def search(self, query, *args, **kwargs):
         """Return set of matches with similarity above 'threshold'.