pdate import path

Author: minghangli-uni

expts_manager/experiment_manager_tool/configuration_updater/common.py

@@ -1,5 +1,5 @@
 
-def update_config_entries(base: dict, change: dict) -> dict:
+def update_config_entries(base: dict, change: dict) -> None:
     """
     Recursively update nuopc_runconfig and config.yaml entries.
     """

expts_manager/experiment_manager_tool/configuration_updater/config_updater.py

@@ -1,11 +1,11 @@
 import os
 import warnings
-from utils.ryaml_handler import read_yaml, write_yaml
-from configuration_updater.common import update_config_entries
+from experiment_manager_tool.utils.ryaml_handler import read_yaml, write_yaml
+from experiment_manager_tool.configuration_updater.common import update_config_entries
 
 class ConfigUpdater:
     @staticmethod
-    def update_config_params(expt_path: str, param_dict: dict, filename):
+    def update_config_params(expt_path: str, param_dict: dict, filename: str) -> None:
         """
         Updates namelist parameters and overwrites namelist file.
 
@@ -29,7 +29,7 @@ def update_config_params(expt_path: str, param_dict: dict, filename):
         param_dict["jobname"] = expt_name
         update_config_entries(file_read, param_dict)
         write_yaml(file_read, nml_path)
-    
+
     @staticmethod
     def update_perturb_jobname(expt_path: str, expt_name: str) -> None:
         """

expts_manager/experiment_manager_tool/configuration_updater/f90nml_updater.py

@@ -1,8 +1,8 @@
 import os
 import numpy as np
 import f90nml
-from utils import constants
-from utils.util_functions import get_namelist_group
+from experiment_manager_tool.utils import constants
+from experiment_manager_tool.utils.util_functions import get_namelist_group
 
 class f90nmlUpdater:
     def __init__(self):

expts_manager/experiment_manager_tool/configuration_updater/mom6_updater.py

@@ -1,6 +1,6 @@
 import os
-from utils.base_manager import BaseManager
-from mixins.mixins import FullPathMixin, OM3UtilsLoaderMixin
+from experiment_manager_tool.utils.base_manager import BaseManager
+from experiment_manager_tool.mixins.mixins import FullPathMixin, OM3UtilsLoaderMixin
 
 class MOM6Updater(BaseManager, FullPathMixin, OM3UtilsLoaderMixin):
     def __init__(self, yamlfile: str) -> None:

expts_manager/experiment_manager_tool/configuration_updater/nuopc_runconfig_updater.py

@@ -1,8 +1,8 @@
 import os
-from mixins.mixins import FullPathMixin, OM3UtilsLoaderMixin
-from configuration_updater.common import update_config_entries
-from utils.base_manager import BaseManager
-from utils.util_functions import get_namelist_group, create_nested_dict
+from experiment_manager_tool.mixins.mixins import FullPathMixin, OM3UtilsLoaderMixin
+from experiment_manager_tool.configuration_updater.common import update_config_entries
+from experiment_manager_tool.utils.base_manager import BaseManager
+from experiment_manager_tool.utils.util_functions import get_namelist_group, create_nested_dict
 
 class NuopcRunConfigUpdater(BaseManager, FullPathMixin, OM3UtilsLoaderMixin):
     def __init__(self, yamlfile: str) -> None:

expts_manager/experiment_manager_tool/main.py

@@ -1,8 +1,6 @@
 #!/usr/bin/env python3
-
 import argparse
-import os
-from manager.experiment_manager import ExperimentManager
+from experiment_manager_tool.manager.experiment_manager import ExperimentManager
 
 def main():
     parser = argparse.ArgumentParser(

expts_manager/experiment_manager_tool/pbs_job_manager/pbs_job_manager.py

@@ -1,7 +1,7 @@
 import os
 import subprocess
 import glob
-from git_manager.git_manager import check_and_commit_changes
+from experiment_manager_tool.git_manager.git_manager import check_and_commit_changes
 
 class PBSJobManager:
     """

expts_manager/experiment_manager_tool/utils/base_manager.py

@@ -1,6 +1,6 @@
 import os
-from utils.ryaml_handler import read_yaml
-from utils import constants
+from experiment_manager_tool.utils.ryaml_handler import read_yaml
+from experiment_manager_tool.utils import constants
 
 class BaseManager:
     """

expts_manager/experiment_manager_tool/utils/load_external_tools.py

@@ -3,8 +3,8 @@
 import warnings
 import shutil
 import subprocess
-from mixins.mixins import FullPathMixin
-from utils.base_manager import BaseManager
+from experiment_manager_tool.mixins.mixins import FullPathMixin
+from experiment_manager_tool.utils.base_manager import BaseManager
 
 class ExternalTools(BaseManager, FullPathMixin):
     def __init__(self, yamlfile: str) -> None:

expts_manager/experiment_manager_tool/utils/metadata.py

@@ -1,5 +1,5 @@
 import os
-from utils.ryaml_handler import read_yaml, write_yaml, LiteralString
+from experiment_manager_tool.utils.ryaml_handler import read_yaml, write_yaml, LiteralString
 
 class MetaData:
     def __init__(self, startfrom_str: str, force_restart: bool, base_path: str, branch_perturb: str, base_branch_name: str) -> None: