qev / load_qev.py
ken-jo's picture
QEV SDK 0.2.0: packaged playground and typed decisions
6cb324a verified
Raw History Blame Contribute Delete
394 Bytes
"""Run the packaged example with the latest bundled qev wheel."""
import json
from pathlib import Path
import torch
from qev import load, DecisionRequest
root = Path(__file__).resolve().parent
torch.set_num_threads(4)
model = load(root)
request = DecisionRequest.from_json((root / "sample_request.json").read_text("utf-8"))
print(json.dumps(model.predict(request, root), indent=2))