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.