change early stopping imports

aramis-lab · Nov 12, 2024 · c1afa03 · c1afa03
1 parent 4b9983b
commit c1afa03
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/clinicadl/commandline/modules_options/early_stopping.py b/clinicadl/commandline/modules_options/early_stopping.py
@@ -2,7 +2,7 @@
 
 from clinicadl.config.config_utils import get_default_from_config_class as get_default
 from clinicadl.config.config_utils import get_type_from_config_class as get_type
-from clinicadl.utils.early_stopping.config import EarlyStoppingConfig
+from clinicadl.optim.early_stopping import EarlyStoppingConfig
 
 # Early Stopping
 patience = click.option(

diff --git a/clinicadl/trainer/config/train.py b/clinicadl/trainer/config/train.py
@@ -17,13 +17,12 @@
 from clinicadl.experiment_manager.config import MapsManagerConfig
 from clinicadl.networks.old_network.config import NetworkConfig
 from clinicadl.optim.config import OptimizationConfig
-from clinicadl.optim.optimizers.config import OptimizerConfig
+from clinicadl.optim.early_stopping import EarlyStoppingConfig
 from clinicadl.predictor.validation import ValidationConfig
 from clinicadl.splitter.config import SplitConfig
 from clinicadl.trainer.transfer_learning import TransferLearningConfig
 from clinicadl.transforms.config import TransformsConfig
 from clinicadl.utils.computational.computational import ComputationalConfig
-from clinicadl.utils.early_stopping.config import EarlyStoppingConfig
 from clinicadl.utils.enum import Task
 
 logger = getLogger("clinicadl.training_config")

diff --git a/clinicadl/trainer/old_trainer.py b/clinicadl/trainer/old_trainer.py
@@ -17,7 +17,7 @@
 
 from clinicadl.splitter.split_utils import find_finished_splits, find_stopped_splits
 from clinicadl.dataset.caps_dataset import return_dataset
-from clinicadl.utils.early_stopping.early_stopping import EarlyStopping
+from clinicadl.optim.early_stopping import EarlyStopping
 from clinicadl.utils.exceptions import MAPSError
 from clinicadl.utils.computational.ddp import DDP
 from clinicadl.utils import cluster
@@ -27,7 +27,6 @@
 from clinicadl.utils.seed import pl_worker_init_function, seed_everything
 from clinicadl.experiment_manager.maps_manager import MapsManager
 from clinicadl.utils.seed import get_seed
-from clinicadl.utils.enum import Task
 from clinicadl.utils.iotools.trainer_utils import (
     create_parameters_dict,
     patch_to_read_json,
@@ -37,7 +36,6 @@
 from clinicadl.predictor.config import PredictConfig
 from clinicadl.splitter.old_splitter import Splitter
 from clinicadl.splitter.config import SplitterConfig
-from clinicadl.transforms.config import TransformsConfig
 
 if TYPE_CHECKING:
     from clinicadl.callbacks.callbacks import Callback