Wrappers

EnsureInfoKeysWrapper

Validates that required keys are present in the info dict after reset and step.

EnsureImageShape

Validates that an image in the info dict has the expected spatial dimensions.

EnsureGoalInfoWrapper

Validates that 'goal' key is present in info dict during reset and/or step.

EverythingToInfoWrapper

Moves all transition information into the info dict for unified data access.

AddPixelsWrapper

Adds rendered environment pixels to info dict with optional resizing and transforms.

ResizeGoalWrapper

Resizes goal images in info dict with optional transforms.

MegaWrapper

Combines multiple wrappers for comprehensive environment preprocessing.

VariationWrapper

Manages variation spaces for vectorized environments.