<gh_stars>0
from envs.race_strategy_full import RaceModel
import numpy as np
import time

mdp = RaceModel(verbose=False, n_cores=6)
_ = mdp.reset()
ret = 0
lap = 1

start = time.time()
while True:
    agent = mdp.get_next_agent()
    #print("Lap {}, agent {}".format(lap, agent))
    a = np.random.choice([0, 1, 2, 3])
    s, r, done, _ = mdp.partial_step(a, agent)
    #print("Reward:" + str(r))
    #mdp.set_signature(mdp.get_signature())
    ret += r
    if mdp.has_transitioned():
        lap += 1
        now = time.time()
        milliseconds = now-start
        print("----------------------------------")
        print("full step time:", milliseconds)
        start = time.time()
    if done:
        print("Return:", ret)
        # print("Race Time:", mdp.time)
        break
