Try using episode._last_infos[agent_id], where agent_id is either the agent_id you want the infos for or taken fromray.rllib.env.base_env import _DUMMY_AGENT_ID. Please open a Github issue if this does not work for you.
1 Like