From e372f953f03455bcd795848551a1646bb852675e Mon Sep 17 00:00:00 2001 From: Artur Niederfahrenhorst Date: Tue, 18 Jul 2023 16:57:13 -0700 Subject: [PATCH] Add missing line to reset obs fn Signed-off-by: Artur Niederfahrenhorst --- rllib/evaluation/env_runner_v2.py | 1 + 1 file changed, 1 insertion(+) diff --git a/rllib/evaluation/env_runner_v2.py b/rllib/evaluation/env_runner_v2.py index e36e00b679142..aaaed3cea6497 100644 --- a/rllib/evaluation/env_runner_v2.py +++ b/rllib/evaluation/env_runner_v2.py @@ -781,6 +781,7 @@ def __process_resetted_obs_for_eval( SampleBatch.NEXT_OBS: obs, SampleBatch.INFOS: infos, SampleBatch.T: episode.length, + SampleBatch.AGENT_INDEX: episode.agent_index(agent_id), }, ) for agent_id, obs in agents_obs