Spaces:
Sleeping
Sleeping
File size: 2,163 Bytes
fa3d628 1380c2c a081d81 fa3d628 a081d81 fa3d628 1380c2c fa3d628 1380c2c a081d81 fa3d628 1380c2c fa3d628 1380c2c fa3d628 1380c2c a081d81 | 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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | """Centralise le chargement de la configuration de préparation du projet."""
from pathlib import Path
import yaml
PROJECT_ROOT = Path(__file__).resolve().parents[1]
DEFAULT_CONFIG_PATH = PROJECT_ROOT / "config" / "project_paths.yaml"
def _resolve_path(raw_path: str) -> Path:
"""Résout un chemin de configuration relativement à la racine du projet."""
path = Path(raw_path)
if path.is_absolute():
return path
return PROJECT_ROOT / path
def ensure_preparation_directories(config: dict[str, object]) -> dict[str, object]:
"""Crée les dossiers cibles attendus par la préparation.
Args:
config: Configuration chargee depuis `project_paths.yaml`.
Returns:
dict[str, object]: Configuration inchangée, pour permettre le chainage.
"""
artifacts_dir = config["ARTIFACTS_DIR"]
pca_artifacts_dir = config["PCA_ARTIFACTS_DIR"]
dataset_path = config["DATASET_PATH"]
if isinstance(artifacts_dir, Path):
artifacts_dir.mkdir(parents=True, exist_ok=True)
if isinstance(pca_artifacts_dir, Path):
pca_artifacts_dir.mkdir(parents=True, exist_ok=True)
if isinstance(dataset_path, Path):
dataset_path.parent.mkdir(parents=True, exist_ok=True)
return config
def load_preparation_config(
config_path: Path | None = None,
*,
ensure_dirs: bool = False,
) -> dict[str, object]:
"""Charge la configuration de préparation depuis le fichier YAML du projet.
Args:
config_path: Chemin optionnel vers un fichier YAML de configuration.
ensure_dirs: Crée les dossiers cibles si `True`.
Returns:
dict[str, object]: Configuration normalisée avec des `Path` résolus.
"""
path = config_path or DEFAULT_CONFIG_PATH
raw_config = yaml.safe_load(path.read_text())
preparation = raw_config["preparation"]
config: dict[str, object] = {}
for key, value in preparation["paths"].items():
config[key] = _resolve_path(value)
for key, value in preparation["parameters"].items():
config[key] = value
if ensure_dirs:
ensure_preparation_directories(config)
return config
|