Skip to content
Snippets Groups Projects
Commit 6355dc02 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[libs.common.data.datamodule] Fix typing errors

parent 74b26aaa
No related branches found
No related tags found
1 merge request!46Create common library
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment