Trajectory Recorder

These are interfaces, classes and utility functions for recording trajectory data:

InMemoryDataset

Base class of trajectory data set for imitation learning that keeps all loaded data in memory.

DataLoadWorker

Data loading worker used to map states to actual observations.

TrajectoryProcessor

Base class for processing individual trajectories.

IdentityTrajectoryProcessor

Identity processing method

DeadEndClippingTrajectoryProcessor

Implementation of the dead-end-clipping preprocessor.

SpacesRecord

Record of spaces (i.e., raw action, observation, and associated data) from a single sub-step.

StepKeyType

The central part of internal API.

StructuredSpacesRecord

Records spaces (i.e., raw actions and observations) from a single environment step.

StateRecord

Keeps trajectory data for one step.

TrajectoryRecord

Common functionality of trajectory records.

StateTrajectoryRecord

Holds state record data (i.e., Maze states and actions, independent of the current

SpacesTrajectoryRecord

Holds structured spaces records (i.e., raw actions and observations recorded during a rollout).

MonitoringSetup

Simple setup for environment monitoring.

SimpleTrajectoryRecordingSetup

Simple setup for trajectory data recording.

TrajectoryWriterRegistry

Handles registration of trajectory data writers.

TrajectoryWriter

Interface for modules serializing the trajectory data.

TrajectoryWriterFile

Simple trajectory data writer.