scripts/estimate_params.py

from pathlib import Path
import sys

ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))

from ares_core.config import AresConfig, estimate_parameters

for cfg_path in sorted((ROOT / "configs").glob("*.json")):
    cfg = AresConfig.from_json(cfg_path)
    print(f"{cfg_path.name}: ~{estimate_parameters(cfg):,} parameters")