TrajectoryWriterFile

class maze.core.trajectory_recording.writers.trajectory_writer_file.TrajectoryWriterFile(log_dir: str | 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: StateTrajectoryRecord) → None

(overrides TrajectoryWriter)

Write episode trajectory data to a file using pickle.

Parameters:

episode_record – Episode trajectory data