imitation.data

Modules handling environment data.

For example: types for transitions/trajectories; methods to compute rollouts; buffers to store transitions; helpers for these modules.

Modules

imitation.data.buffer	Buffers to store NumPy arrays and transitions in.
imitation.data.huggingface_utils	Helpers to convert between Trajectories and HuggingFace's datasets library.
imitation.data.rollout	Methods to collect, analyze and manipulate transition and trajectory rollouts.
imitation.data.serialize	Serialization utilities for trajectories.
imitation.data.types	Types and helper methods for transitions and trajectories.
imitation.data.wrappers	Environment wrappers for collecting rollouts.