cli

Author: PhilipJohnBasile

src/cardvalue_ml/models/experiments.py

@@ -3,17 +3,19 @@
 from __future__ import annotations
 
 import datetime as dt
+import importlib
 from pathlib import Path
-from typing import Any
+from typing import Any, cast
 
 import pandas as pd
-import yaml  # type: ignore[import-untyped]
 
 from cardvalue_ml.config import paths
 from cardvalue_ml.features.build_features import prepare_regression_features
 from cardvalue_ml.models.evaluate import save_json
 from cardvalue_ml.models.train import train_random_forest
 
+yaml = cast(Any, importlib.import_module("yaml"))
+
 
 def load_experiment_config(config_path: Path) -> dict[str, Any]:
     if not config_path.exists():
@@ -22,7 +24,7 @@ def load_experiment_config(config_path: Path) -> dict[str, Any]:
         config = yaml.safe_load(handle)
     if not isinstance(config, dict):
         raise ValueError("Experiment config must be a mapping")
-    return config
+    return cast(dict[str, Any], config)
 
 
 def run_experiment(config_path: Path) -> dict[str, Any]: