@@ -32,15 +32,13 @@ def load_experiment_cfg(experiment_name: str, ckpt_root_dir: str = None) -> Dict
 
     resume_dir = Path(checkpoints_dir_path) / ".hydra"
     if not resume_dir.exists():
-        raise FileNotFoundError(
-            f"The checkpoint directory {checkpoints_dir_path} does not include .hydra artifacts to resume the experiment."
-        )
+        raise FileNotFoundError(f"The checkpoint directory {checkpoints_dir_path} does not include .hydra artifacts to resume the experiment.")
 
     # Load overrides that were used in previous run
     overrides_cfg = list(OmegaConf.load(resume_dir / "overrides.yaml"))
 
     GlobalHydra.instance().clear()
-    with initialize_config_dir(config_dir=normalize_path(str(resume_dir))):
+    with initialize_config_dir(config_dir=normalize_path(str(resume_dir)), version_base="1.2"):
         cfg = compose(config_name="config.yaml", overrides=overrides_cfg)
     return cfg