fix: raise explicit error if config is not found

IamGianluca · Mar 23, 2024 · 529d97b · 529d97b
1 parent 0170e0b
commit 529d97b
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/blazingai/params.py b/blazingai/params.py
@@ -5,8 +5,12 @@
 
 
 def load_cfg(fpath: Path, cfg_name: str) -> DictConfig:
-    cfg = load_hparams_from_yaml(config_yaml=fpath, use_omegaconf=True)
-    cfg = cfg.get(cfg_name)
-    if hasattr(cfg, "lr"):
+    all_configs = load_hparams_from_yaml(config_yaml=fpath, use_omegaconf=True)
+    cfg = all_configs.get(cfg_name)
+    if cfg is None:
+        raise ValueError(f"Could not find config {cfg_name} in {fpath}")
+
+    if hasattr(cfg, "lr"):  # convert exponential notation to float
         cfg.lr = float(cfg.lr)  # type: ignore
+
     return cfg