refactoring

image_prediction/stitcher/stitcher.py

@@ -50,4 +50,7 @@ class Stitcher:
         return pairs
 
     def stitch(self, pairs: Iterable[ImageMetadataPair]) -> List[ImageMetadataPair]:
-        return until_convergence(self.merge_along_both_axes, pairs)
+        def break_condition(pairs1, pairs2):
+            return len(pairs1) == len(pairs2)
+
+        return until_convergence(self.merge_along_both_axes, break_condition, pairs)

image_prediction/stitcher/utils.py

@@ -130,8 +130,12 @@ def make_merger_aggregator(direction):
 
 
 def merge_group(group, direction):
+
+    def break_condition(pairs1, pairs2):
+        return len(pairs1) == len(pairs2)
+
     reduce_group = make_merger_aggregator(direction)
-    return until_convergence(reduce_group, group)
+    return until_convergence(reduce_group, break_condition, group)
 
 
 def merge_group_horizontally(group):

image_prediction/utils/generic.py

@@ -8,7 +8,7 @@ def chunk_iterable(iterable, chunk_size):
     return takewhile(truth, map(tuple, starmap(islice, repeat((iter(iterable), chunk_size)))))
 
 
-def until_convergence(func, *args, **kwargs):
+def until_convergence(func, cond, *args, **kwargs):
     for a, b in chunk_iterable(iterate(func, *args, **kwargs), chunk_size=2):
-        if len(a) == len(b):
+        if cond(a, b):
             return a