Added StopIteration catch

WildMeOrg · Jan 28, 2021 · a0a327c · a0a327c
1 parent 243b8ef
commit a0a327c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/wbia/algo/graph/mixin_loops.py b/wbia/algo/graph/mixin_loops.py
@@ -109,8 +109,11 @@ def main_gen(infr, max_loops=None, use_refresh=True):
         if infr.params['redun.enabled'] and infr.params['redun.enforce_pos']:
             infr.loop_phase = 'pos_redun_init'
             # Fix positive redundancy of anything within the loop
-            for _ in infr.pos_redun_gen():
-                yield _
+            try:
+                for _ in infr.pos_redun_gen():
+                    yield _
+            except StopIteration:
+                pass
 
         infr.phase = 1
         if infr.params['ranking.enabled']: