[RLlib] rollout batch, handle rewards that are None (unknown) in a multi-agent env (#11858) (#11911)

2026-08-11 11:24:51 +08:00 · 2020-11-25 13:39:22 +01:00
parent c009c178f6
commit 09d5413f70
1 changed files with 2 additions and 1 deletions
@@ -451,7 +451,8 @@ def rollout(agent,

            if multiagent:
                done = done["__all__"]
-                reward_total += sum(reward.values())
+                reward_total += sum(
+                    r for r in reward.values() if r is not None)
            else:
                reward_total += reward
            if not no_render: