Skip to content

Commit e30ac95

Browse files
committed
fix: preserve Compose attributes in flatten()
Forward map_items, unpack_items, log_stats, lazy, and overrides to the flattened Compose so the returned pipeline is equivalent to the original, mirroring OneOf.flatten behavior.
1 parent 91f9392 commit e30ac95

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

monai/transforms/compose.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -358,7 +358,14 @@ def flatten(self):
358358
else:
359359
new_transforms.append(t)
360360

361-
return Compose(new_transforms)
361+
return Compose(
362+
new_transforms,
363+
map_items=self.map_items,
364+
unpack_items=self.unpack_items,
365+
log_stats=self.log_stats,
366+
lazy=self._lazy,
367+
overrides=self.overrides,
368+
)
362369

363370
def __len__(self):
364371
"""Return number of transformations."""

0 commit comments

Comments
 (0)