CoolFace
Apppublic

Harshitjhamb/Meta_Hackathon

sourceHugging Faceupdated 5mo agoView on Hugging Face
0likes
test.py12 linesDownload Raw Back to root
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)