Source code for src.gridmind.wrappers.env_wrappers.base_gym_wrapper

import gymnasium


[docs]class BaseGymWrapper:
    def __init__(self, env: gymnasium.Env):
[docs]        self.env = env

[docs]    def reset(self):
        return self.env.reset()

[docs]    def step(self, action):
        return self.env.step(action)

[docs]    def close(self):
        self.env.close()

[docs]    def render(self):
        return self.env.render()

[docs]    def seed(self, seed=None):
        return self.env.seed(seed)

[docs]    def get_action_space(self):
        return self.env.action_space

[docs]    def get_observation_space(self):
        return self.env.observation_space

[docs]    def get_reward_range(self):
        return self.env.reward_range

[docs]    def get_metadata(self):
        return self.env.metadata

[docs]    def get_env(self):
        return self.env

[docs]    def __getattr__(self, name):
        return getattr(self.env, name)