Add seed to dict only if transformed Subject

fepegar · Jul 11, 2020 · 7fbd4af · 7fbd4af
1 parent 65f9e99
commit 7fbd4af
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/torchio/transforms/augmentation/random_transform.py b/torchio/transforms/augmentation/random_transform.py
@@ -32,7 +32,12 @@ def __call__(self, sample: Subject):
         with torch.random.fork_rng():
             self.check_seed()
             transformed = super().__call__(sample)
-            random_params_dict = transformed.history[-1][1]
+        if transformed is sample:
+            pass  # the transform was not applied
+        elif not isinstance(transformed, Subject):
+            pass  # random parameters are stored in instances of Subject
+        else:
+            _, random_params_dict = transformed.history[-1]
             random_params_dict['seed'] = self._seed
         return transformed