Fix bug in datamodule (#148)

creatis-myriad · Nov 21, 2023 · e357b56 · e357b56
1 parent 204d50a
commit e357b56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ascent/datamodules/nnunet_datamodule.py b/ascent/datamodules/nnunet_datamodule.py
@@ -342,7 +342,7 @@ def setup_transforms(self) -> None:
                 other_transforms.append(aug)
 
         if self.hparams.augmentation.get("intensity"):
-            for aug in self.hparams.augmentation.get("intensity"):
+            for name, aug in self.hparams.augmentation.get("intensity").items():
                 other_transforms.append(aug)
 
         if self.hparams.augmentation.get("flip"):