hive.envs.wrappers.gym_wrappers module

class hive.envs.wrappers.gym_wrappers.FlattenWrapper(env)[source]

Bases: gym.core.ObservationWrapper

Flatten the observation to one dimensional vector.

observation(obs)[source]
class hive.envs.wrappers.gym_wrappers.PermuteImageWrapper(env)[source]

Bases: gym.core.ObservationWrapper

Changes the image format from HWC to CHW

observation(obs)[source]