@@ -427,6 +427,7 @@ def get_groups(blocks):
group = tuple(scan_group(blocks, i, block, visited))
for j in group:
groups[j] = groupid
+ groupsizes[j] = len(group)
return groups, groupsizes, groupid