Merge remote-tracking branch 'origin/master'

0b50abd8 · MasterScrat · 94e07ef9 · 96f300bf · 0b50abd8
Commit 0b50abd8 authored May 01, 2020 by MasterScrat
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

envs/flatland/rllib_wrapper.py envs/flatland/rllib_wrapper.py +3 -3

No files found.
--- a/envs/flatland/rllib_wrapper.py
+++ b/envs/flatland/rllib_wrapper.py
@@ -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]