xsx001's picture
Add StateShiftBench review code
89e75e1 verified
#!/usr/bin/env python3
"""Run one StateShiftBench case with the direct reference policy."""
import json
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
from stateshiftbench.runner import run_case
def main():
if len(sys.argv) != 2:
raise SystemExit("usage: python examples/run_one_case.py <case.json>")
case = json.loads(Path(sys.argv[1]).read_text())
print(json.dumps(run_case(case, strategy="direct"), indent=2, sort_keys=True))
if __name__ == "__main__":
main()