diff --git a/src/mednet/libs/common/data/datamodule.py b/src/mednet/libs/common/data/datamodule.py index dff5a24f0b0a95b2c9c1b36e1f962da14da32e4c..4439b9901101170018ad93750ac0fe704b097f63 100644 --- a/src/mednet/libs/common/data/datamodule.py +++ b/src/mednet/libs/common/data/datamodule.py @@ -510,7 +510,10 @@ class ConcatDataModule(lightning.LightningDataModule): @model_transforms.setter def model_transforms(self, value: list[Transform] | None): old_value = self._model_transforms - self._model_transforms = value + if value is None: + self._model_transforms = value + else: + self._model_transforms = list(value) # datasets that have been setup() for the current stage are reset if value != old_value and len(self._datasets):