Skip to content

Commit

Permalink
Fix sb3 tutorials typo (#1133)
Browse files Browse the repository at this point in the history
Co-authored-by: Fernando <[email protected]>
  • Loading branch information
elliottower and Fernadoo authored Nov 15, 2023
1 parent f27e84b commit 2b65b5f
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions tutorials/SB3/kaz/sb3_kaz_vector.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,8 +98,8 @@ def eval(env_fn, num_games: int = 100, render_mode: str | None = None, **env_kwa
for agent in env.agent_iter():
obs, reward, termination, truncation, info = env.last()

for agent in env.agents:
rewards[agent] += env.rewards[agent]
for a in env.agents:
rewards[a] += env.rewards[a]

if termination or truncation:
break
Expand Down
4 changes: 2 additions & 2 deletions tutorials/SB3/pistonball/sb3_pistonball_vector.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,8 +99,8 @@ def eval(env_fn, num_games: int = 100, render_mode: str | None = None, **env_kwa
obs, reward, termination, truncation, info = env.last()

if termination or truncation:
for agent in env.agents:
rewards[agent] += env.rewards[agent]
for a in env.agents:
rewards[a] += env.rewards[a]
break
else:
act = model.predict(obs, deterministic=True)[0]
Expand Down
4 changes: 2 additions & 2 deletions tutorials/SB3/waterworld/sb3_waterworld_vector.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,8 +78,8 @@ def eval(env_fn, num_games: int = 100, render_mode: str | None = None, **env_kwa
for agent in env.agent_iter():
obs, reward, termination, truncation, info = env.last()

for agent in env.agents:
rewards[agent] += env.rewards[agent]
for a in env.agents:
rewards[a] += env.rewards[a]
if termination or truncation:
break
else:
Expand Down

0 comments on commit 2b65b5f

Please sign in to comment.