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]")