ray.rllib.env.single_agent_episode.SingleAgentEpisode.get_data_dict#
- SingleAgentEpisode.get_data_dict()[source]#
Converts a SingleAgentEpisode into a data dict mapping str keys to data.
The keys used are: Columns.EPS_ID, T, OBS, INFOS, ACTIONS, REWARDS, TERMINATEDS, TRUNCATEDS, and those in
self.extra_model_outputs
.- Returns:
A data dict mapping str keys to data records.