Skip to content

Commit

Permalink
Update test
Browse files Browse the repository at this point in the history
  • Loading branch information
jagdeepsb committed Jun 13, 2024
1 parent 96522cc commit c4ec065
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_render.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,10 +13,10 @@ def test_it(self):

for _ in range(100):
action = env.action_space.sample() - 1
ob, reward, done, info = env.step(action)
ob, reward, terminated, truncated, info = env.step(action)
env.render()

if done:
if terminated or truncated:
env.reset()

env.close()

0 comments on commit c4ec065

Please sign in to comment.