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