Commit 905c401b authored by metataro's avatar metataro

do not check "action_required" but query agents each step (prevent missing...

do not check "action_required" but query agents each step (prevent missing rewards for blocking agents)
parent 546f171e
......@@ -53,9 +53,9 @@ class FlatlandRllibWrapper(object):
if agent != '__all__':
if done:
self._agents_done.append(agent)
if infos['action_required'][agent] or done: # filter out agents that need no action
o[agent] = obs[agent]
r[agent] = rewards[agent]
# if infos['action_required'][agent] or done:
o[agent] = obs[agent]
r[agent] = rewards[agent]
self._agent_scores[agent] += rewards[agent]
self._agent_steps[agent] += 1
d[agent] = dones[agent]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment