Fix randomness for threading

Project-MONAI · Jul 17, 2024 · fa4ceb7 · fa4ceb7
1 parent 4fbe800
commit fa4ceb7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/monai/transforms/compose.py b/monai/transforms/compose.py
@@ -107,7 +107,11 @@ def execute_compose(
 
  for _transform in transforms[start:end]:
  if threading:
- _transform = deepcopy(_transform) if isinstance(_transform, ThreadUnsafe) else _transform
+ if isinstance(_transform, ThreadUnsafe):
+ if isinstance(_transform, Randomizable):
+ # update the random state before deepcopy, otherwise there is no randomness
+ _transform.randomize()
+ _transform = deepcopy(_transform)
  data = apply_transform(
  _transform, data, map_items, unpack_items, lazy=lazy, overrides=overrides, log_stats=log_stats
  )