EpisodeRecorder

class maze.core.rollout.parallel_rollout_runner.EpisodeRecorder

Keeps the statistics and event logs from the last episode so that it can then be shipped to the main process.

get_last_episode_data() → EpisodeStatsReport

Get the stats and event log from the last episode.

Returns:

Tuple of (episode stats, event log)

receive(stat: Dict[LogStatsKey, int | float | numpy.ndarray | dict]) → None

(overrides LogStatsConsumer)

Receive the statistics from the env and store them.

write(episode_event_log: EpisodeEventLog) → None

(overrides LogEventsWriter)

Receive the event logs from the env and store them.