Skip to content

Commit

Permalink
Clean up redundant line in parallel_api_test (Farama-Foundation#1095)
Browse files Browse the repository at this point in the history
  • Loading branch information
chrisyeh96 authored Sep 10, 2023
1 parent 63722dd commit 4f3e1f2
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions pettingzoo/test/parallel_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,14 +77,13 @@ def parallel_api_test(par_env: ParallelEnv, num_cycles=1000):
assert isinstance(truncated, dict)
assert isinstance(info, dict)

agents_set = set(live_agents)
keys = "observation reward terminated truncated info".split()
vals = [obs, rew, terminated, truncated, info]
for k, v in zip(keys, vals):
key_set = set(v.keys())
if key_set == agents_set:
if key_set == live_agents:
continue
if len(key_set) < len(agents_set):
if len(key_set) < len(live_agents):
warnings.warn(f"Live agent was not given {k}")
else:
warnings.warn(f"Agent was given {k} but was dead last turn")
Expand Down

0 comments on commit 4f3e1f2

Please sign in to comment.