diff --git a/realhf/apps/quickstart.py b/realhf/apps/quickstart.py index 9a28b6a..a3fdfd4 100644 --- a/realhf/apps/quickstart.py +++ b/realhf/apps/quickstart.py @@ -64,8 +64,8 @@ def main(): launch_hydra_task(args["cmd"], experiment_name, trial_name, QUICKSTART_FN[args["cmd"]]) -def prepare_hydra_config(name: str, PROLOGUE_path: str): - config = OmegaConf.load(PROLOGUE_path) +def prepare_hydra_config(name: str, prologue_path: str): + config = OmegaConf.load(prologue_path) experiment_name = get_experiment_name(config.get("experiment_name")) trial_name = get_trial_name(config.get("trial_name")) config_dir = f"{cluster_spec.fileroot}/configs/{getpass.getuser()}/{experiment_name}/{trial_name}" diff --git a/realhf/base/prologue.py b/realhf/base/prologue.py index 76c602d..f0a2242 100644 --- a/realhf/base/prologue.py +++ b/realhf/base/prologue.py @@ -19,9 +19,9 @@ def global_init(): args = vars(parser.parse_known_args()[0]) if args[PROLOGUE_FLAG_VAR_NAME] is None: return - PROLOGUE_path = args[PROLOGUE_FLAG_VAR_NAME] + prologue_path = args[PROLOGUE_FLAG_VAR_NAME] - config = OmegaConf.load(PROLOGUE_path) + config = OmegaConf.load(prologue_path) external_configs = config.get(PROLOGUE_EXTERNAL_CONFIG_NAME) if external_configs is None: