File size: 1,001 Bytes
c8d9552 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | """Runnable quickstart for the SnapJudge HF repo."""
from pathlib import Path
from snapjudge.agent import load
from snapjudge.router import GameRouter
from snapjudge.data_gen import questions_for
_HERE = Path(__file__).resolve().parent
router = GameRouter(agent=load(str(_HERE))) # robust to cwd
demos = [
({"game": "tictactoe", "board": ["X", "X", " ", "O", "O", " ", " ", " ", " "],
"player": "X", "board_str": "XX.OO...."},
questions_for("tictactoe"), "tic-tac-toe"),
({"game": "snake", "head": [5, 5], "body": [[5, 5], [5, 6]], "food": [7, 5], "grid": [10, 10]},
questions_for("snake"), "snake"),
({"game": "templerun", "obstacle": "gap", "lane": "mid", "distance": "near", "speed": "fast"},
questions_for("templerun"), "temple-run"),
]
for state, qs, title in demos:
r = router.predict(state, qs)
print(f"=== {title} (route={r['routing']['model']}: {r['routing']['reason']}) ===")
for qid, a in r["answers"].items():
print(f" {qid}: {a}")
|