Add close. Fix rendering to conform with gym
This commit is contained in:
parent
62fecda9d6
commit
bedb4297f2
@ -7,8 +7,6 @@ from fancy_gym.envs.mujoco.air_hockey import position_control_wrapper as positio
|
||||
from fancy_gym.envs.mujoco.air_hockey.utils import robot_to_world
|
||||
from mushroom_rl.core import Environment
|
||||
|
||||
MAX_EPISODE_STEPS_AIRHOCKEY = 45000 # For a tournament env, the game can last up to 15 minutes
|
||||
|
||||
class AirHockeyEnv(Environment):
|
||||
metadata = {"render_modes": ["human"], "render_fps": 60}
|
||||
|
||||
@ -84,6 +82,8 @@ class AirHockeyEnv(Environment):
|
||||
self.env_info['env_name'] = self.env_name
|
||||
|
||||
self.render_mode = render_mode
|
||||
self.render_human_active = False
|
||||
|
||||
super().__init__(self.base_env.info)
|
||||
|
||||
def step(self, action):
|
||||
@ -111,10 +111,14 @@ class AirHockeyEnv(Environment):
|
||||
|
||||
if self.env_info['env_name'] == "tournament":
|
||||
obs = np.array(np.split(obs, 2))
|
||||
|
||||
if self.render_human_active:
|
||||
self.base_env.render()
|
||||
|
||||
return obs, reward, done, False, info
|
||||
|
||||
def render(self):
|
||||
self.base_env.render()
|
||||
self.render_human_active = True
|
||||
|
||||
def reset(self, seed=None, options={}):
|
||||
self.base_env.seed(seed)
|
||||
@ -146,6 +150,9 @@ class AirHockeyEnv(Environment):
|
||||
@property
|
||||
def unwrapped(self):
|
||||
return self
|
||||
|
||||
def close(self):
|
||||
return
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
Loading…
Reference in New Issue
Block a user