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
         )