#! /usr/local/bin/python3
import os
import json

_train=True
_inference=True
from train import *
os.environ["SM_MODEL_DIR"]="./mock/out"

if _train:
    train({
    },{
        
    },[],0,out_dir=os.environ["SM_MODEL_DIR"],current_host="local")

if _inference:
    from host import *
    model=model_fn("./tmp/out")
    width=10
    height=10
    
    body=json.dumps({
        "opponent":"test",
        "board":[[0]*width for x in range(height)],
    })
    response=transform_fn(model,body,"application/json","application/json")
    print(response)