TrajectoryWriterFile

class maze.core.trajectory_recording.writers.trajectory_writer_file.TrajectoryWriterFile(log_dir: Union[str, pathlib.Path] = PosixPath('trajectory_data'))

Simple trajectory data writer. Serializes trajectory data for each episode into a separate file using Pickle.

Suitable for smaller scale rollouts or debugging.

Parameters

log_dir – Where trajectory data should be logged.

write(episode_record: maze.core.trajectory_recording.records.trajectory_record.StateTrajectoryRecord) → None

(overrides TrajectoryWriter)

Write episode trajectory data to a file using pickle.

Parameters

episode_record – Episode trajectory data