remove split list

aramis-lab · Oct 14, 2024 · 0f069af · 0f069af
1 parent feff9ec
commit 0f069af
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/clinicadl/trainer/trainer.py b/clinicadl/trainer/trainer.py
@@ -220,7 +220,7 @@ def train(
 
         else:
             splitter_config = SplitterConfig(**self.config.get_dict())
-            split_manager = Splitter(splitter_config, split_list=split_list)
+            split_manager = Splitter(splitter_config)
 
             for split in split_manager.split_iterator():
                 logger.info(f"Training split {split}")
@@ -244,7 +244,7 @@ def train(
     def check_split_list(self, split_list, overwrite):
         existing_splits = []
         splitter_config = SplitterConfig(**self.config.get_dict())
-        split_manager = Splitter(splitter_config, split_list=split_list)
+        split_manager = Splitter(splitter_config)
         for split in split_manager.split_iterator():
             split_path = self.maps_manager.maps_path / f"split-{split}"
             if split_path.is_dir():
@@ -281,7 +281,7 @@ def _resume(
         """
         missing_splits = []
         splitter_config = SplitterConfig(**self.config.get_dict())
-        split_manager = Splitter(splitter_config, split_list=split_list)
+        split_manager = Splitter(splitter_config)
         for split in split_manager.split_iterator():
             if not (self.maps_manager.maps_path / f"split-{split}" / "tmp").is_dir():
                 missing_splits.append(split)
@@ -473,8 +473,8 @@ def _train_ssda(
 
         splitter_config = SplitterConfig(**self.config.get_dict())
 
-        split_manager = Splitter(splitter_config, split_list=split_list)
-        split_manager_target_lab = Splitter(splitter_config, split_list=split_list)
+        split_manager = Splitter(splitter_config)
+        split_manager_target_lab = Splitter(splitter_config)
 
         for split in split_manager.split_iterator():
             logger.info(f"Training split {split}")