File size: 865 Bytes
6cda091
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
"""
Utilidad para cargar archivos de configuración YAML.
Todos los módulos deben usar esto en lugar de hardcodear valores.
"""
import yaml
from pathlib import Path


def load_config(config_path: str) -> dict:
    """
    Carga un archivo YAML de configuración.

    Args:
        config_path: Ruta al archivo YAML (relativa a la raíz del proyecto).

    Returns:
        Diccionario con la configuración.
    """
    path = Path(config_path)
    if not path.exists():
        raise FileNotFoundError(f"Config file not found: {config_path}")

    with open(path, "r") as f:
        return yaml.safe_load(f)


def load_pipeline_config() -> dict:
    return load_config("configs/pipeline.yaml")


def load_features_config() -> dict:
    return load_config("configs/features.yaml")


def load_models_config() -> dict:
    return load_config("configs/models.yaml")