Fzina commited on
Commit
26e1a85
·
verified ·
1 Parent(s): 91028ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -147,7 +147,7 @@ def optimize_signal_rl(congestion_level):
147
  model.learn(total_timesteps=1000)
148
 
149
  # Reset environment to start the simulation
150
- obs, _ = env.reset()
151
 
152
  # Loop through to simulate for 10 timesteps
153
  for _ in range(10):
 
147
  model.learn(total_timesteps=1000)
148
 
149
  # Reset environment to start the simulation
150
+ obs, info = env.reset()
151
 
152
  # Loop through to simulate for 10 timesteps
153
  for _ in range(10):