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): def configure(self, args):
pass pass
if parse_version(gym.__version__)>=parse_version('0.9.6'): if parse_version(gym.__version__) < parse_version('0.9.6'):
close = _close _render = render
render = _render _reset = reset
reset = _reset _seed = seed
seed = _seed _step = step
step = _step _close = close
@property @property

View File

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

View File

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