Joined subsequent for loops to avoid keeping faces in memory.

dask · Oct 19, 2023 · 43efb6a · 43efb6a
1 parent 56ac712
commit 43efb6a
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/dask_image/ndmeasure/_utils/_label.py b/dask_image/ndmeasure/_utils/_label.py
@@ -164,12 +164,9 @@ def label_adjacency_graph(labels, structure, nlabels, wrap_axes=None):
         labels.chunks, labels.shape, structure, wrap_axes=wrap_axes
     )
     all_mappings = [da.empty((2, 0), dtype=LABEL_DTYPE, chunks=1)]
-    faces = []
 
     for face_slice in face_slices:
-        faces.append(labels[face_slice])
-
-    for face in faces:
+        face = labels[face_slice]
         mapped = _across_block_label_grouping_delayed(face, structure)
         all_mappings.append(mapped)