From 6355dc027b47f703dbf8d19141831a71cb4a72ad Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 25 Jun 2024 15:47:26 +0200
Subject: [PATCH] [libs.common.data.datamodule] Fix typing errors
---
src/mednet/libs/common/data/datamodule.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/src/mednet/libs/common/data/datamodule.py b/src/mednet/libs/common/data/datamodule.py
index dff5a24f..4439b990 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):
--
GitLab