SPB Git

spb/wp24_uqo

Public
886 B · 18 lines python
1"""Solve one scenario of the two-class model. Usage: python3 scripts/20_solve_scenario.py <name> [T]2Scenario definitions live in src/agi_ge/scenarios.py; results go to results/scenarios/."""3import json4import os5import sys67sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "src"))8from agi_ge.scenarios import SCENARIOS  # noqa: E4029from agi_ge.solve import solve_scenario  # noqa: E4021011name = sys.argv[1]12T = int(sys.argv[2]) if len(sys.argv) > 2 else 27513sc = SCENARIOS[name]14outdir = os.path.join(os.path.dirname(__file__), "..", "results", "scenarios")15p, m, u, P, out = solve_scenario(name, overrides=sc.get("overrides"), param_overrides=sc.get("params"), T=T,16                                 outdir=outdir, **sc.get("solver", {}))17print(json.dumps({k: out[k] for k in ("name", "converged", "final_res", "seconds", "horizon_check")}, indent=1, default=float))18
Files · cdaafa84383ba9e74ff46735fd4b4598b964418c
Menu

Theme