ThomasSimonini HF staff commited on
Commit
e0898c0
1 Parent(s): 197c70e

Change max steps

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def replay(env_name, max_steps, time_sleep):
42
  done = False
43
  i = 0
44
  while not done:
45
- i++
46
  if i < max_steps:
47
  frame = env.render(mode="rgb_array")
48
  action, _states = model.predict(obs)
 
42
  done = False
43
  i = 0
44
  while not done:
45
+ i+= 1
46
  if i < max_steps:
47
  frame = env.render(mode="rgb_array")
48
  action, _states = model.predict(obs)