File size: 347 Bytes
4bbf0fc | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | from env import TrafficEnv
from tasks import EASY_CONFIG
env = TrafficEnv(EASY_CONFIG)
print("[START]")
state = env.reset()
done = False
step_count = 0
while not done:
action = 0
next_state, reward, done, info = env.step(action)
print(f"[STEP] step={step_count}, reward={reward}, done={done}")
step_count += 1
print("[END]")
|