examples/quickstart.py

"""Score a released example: python examples/quickstart.py --game snake."""

import argparse
import json
from pathlib import Path
import sys

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

parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--game", choices=["dino", "snake", "tetris"], default="snake")
parser.add_argument("--device", default=None)
args = parser.parse_args()
row = json.loads((ROOT / args.game / "example.json").read_text())
model = GameModel.from_pretrained(ROOT, game=args.game, device=args.device)
print(json.dumps(model.predict(row["state"], row["question"]), indent=2))