diff --git a/clinicadl/utils/caps_dataset/data.py b/clinicadl/utils/caps_dataset/data.py index 1c0dc74e3..01174051e 100644 --- a/clinicadl/utils/caps_dataset/data.py +++ b/clinicadl/utils/caps_dataset/data.py @@ -224,9 +224,9 @@ def _get_meta_data(self, idx: int) -> Tuple[str, str, str, int, int]: label = self.label_fn(target) domain = self.df.loc[image_idx, "domain"] # TO CHECK domain = self.domain_fn(domain) - else: label = -1 + domain = None return participant, session, cohort, elem_idx, label, domain