Fix formating issues

This commit is contained in:
Antonin RAFFIN
2018-12-31 12:48:42 +01:00
parent 0df3527884
commit 1c61b629e3
3 changed files with 24 additions and 28 deletions

View File

@@ -250,12 +250,12 @@ class HumanoidDeepMimicGymEnv(gym.Env):
def configure(self, args):
pass
if parse_version(gym.__version__)>=parse_version('0.9.6'):
close = _close
render = _render
reset = _reset
seed = _seed
step = _step
if parse_version(gym.__version__) < parse_version('0.9.6'):
_render = render
_reset = reset
_seed = seed
_step = step
_close = close
@property

View File

@@ -69,7 +69,7 @@ class MJCFBaseBulletEnv(gym.Env):
self.potential = self.robot.calc_potential()
return s
def render(self, mode, close=False):
def render(self, mode='human'):
if mode == "human":
self.isRender = True
if mode != "rgb_array":

View File

@@ -196,8 +196,6 @@ class PyBulletSimGymEnv(gym.Env):
rgb_array = rgb_array[:, :, :3]
return rgb_array
def _termination(self):
terminate=self._example_sim.Termination()
return terminate
@@ -206,14 +204,12 @@ class PyBulletSimGymEnv(gym.Env):
reward = 0
return reward
def _get_observation(self):
self._observation = self._example_sim.GetObservation()
return self._observation
if parse_version(gym.__version__)>=parse_version('0.9.6'):
render = _render
reset = _reset
seed = _seed
step = _step
if parse_version(gym.__version__) < parse_version('0.9.6'):
_render = render
_reset = reset
_seed = seed
_step = step