from __future__ import annotations
import os
from pathlib import Path

ROOT = Path(os.environ.get("RIEMANN_ROOT", "C:/RiemannModel"))
DATA_DIR = ROOT / "data"
CHECKPOINT_ROOT = ROOT / "checkpoints"
CONTROLLER_ROOT = ROOT / "ntk_controllers"
HF_CACHE_DIR = str(ROOT / "hf_cache")