Skip to content

Commit fc193b6

Browse files
Fix randomness for threading
Signed-off-by: marcus.wirtz <[email protected]>
1 parent 4fbe800 commit fc193b6

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

monai/transforms/compose.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,11 @@ def execute_compose(
107107

108108
for _transform in transforms[start:end]:
109109
if threading:
110-
_transform = deepcopy(_transform) if isinstance(_transform, ThreadUnsafe) else _transform
110+
if isinstance(_transform, ThreadUnsafe):
111+
if isinstance(_transform, Randomizable):
112+
# update the random state before deepcopy, otherwise there is no randomness
113+
_transform.randomize()
114+
_transform = deepcopy(_transform)
111115
data = apply_transform(
112116
_transform, data, map_items, unpack_items, lazy=lazy, overrides=overrides, log_stats=log_stats
113117
)

0 commit comments

Comments
 (0)