pantheonrl.common.wrappers

Collection of environment wrappers.

This module implements the FrameStack and Recording wrappers for TurnBasedEnv and SimultaneousEnv

Functions

frame_wrap

Construct FrameStack environment for the given env

recorder_wrap

Construct a Recorder environment for the given env

Classes

HistoryQueue

Ring buffer representing the saved history for the FrameStack wrappers.

MultiRecorder

Base Class for all Recorder Wrappers

SimultaneousFrameStack

Wrapper that stacks the observations of a simultaneous environment.

SimultaneousRecorder

Recorder for all turn-based environments

TurnBasedFrameStack

Wrapper that stacks the observations of a turn-based environment.

TurnBasedRecorder

Recorder for all turn-based environments