Harshitjhamb/Meta_Hackathon
0
1from env import TrafficEnv2from models import Action3env = TrafficEnv()4obs = env.reset()5print("Initial:", obs)6for i in range(10):7 action = Action(signal=i % 4) # switch lanes8 result = env.step(action)9 print(f"\nStep {i+1}")10 print("Lanes:", result.observation.lanes)11 print("Signal:", result.observation.current_signal)12 print("Reward:", result.reward)