Merge pull request #1547 from wenkesj/master

pybullet support for gym BREAKING CHANGES, including v0.9.x
This commit is contained in:
erwincoumans
2018-02-05 09:55:26 -08:00
committed by GitHub
9 changed files with 140 additions and 97 deletions

View File

@@ -99,3 +99,8 @@ class CartPoleBulletEnv(gym.Env):
def _render(self, mode='human', close=False): def _render(self, mode='human', close=False):
return return
render = _render
reset = _reset
seed = _seed
step = _step

View File

@@ -254,3 +254,7 @@ class KukaCamGymEnv(gym.Env):
#print(reward) #print(reward)
return reward return reward
render = _render
reset = _reset
seed = _seed
step = _step

View File

@@ -276,10 +276,7 @@ class KukaGymEnv(gym.Env):
#print(reward) #print(reward)
return reward return reward
def reset(self): render = _render
"""Resets the state of the environment and returns an initial observation. reset = _reset
seed = _seed
Returns: observation (object): the initial observation of the step = _step
space.
"""
return self._reset()

View File

@@ -385,3 +385,8 @@ class MinitaurBulletDuckEnv(gym.Env):
scale=self._observation_noise_stdev, size=observation.shape) * scale=self._observation_noise_stdev, size=observation.shape) *
self.minitaur.GetObservationUpperBound()) self.minitaur.GetObservationUpperBound())
return observation return observation
render = _render
reset = _reset
seed = _seed
step = _step

View File

@@ -376,3 +376,8 @@ class MinitaurBulletEnv(gym.Env):
scale=self._observation_noise_stdev, size=observation.shape) * scale=self._observation_noise_stdev, size=observation.shape) *
self.minitaur.GetObservationUpperBound()) self.minitaur.GetObservationUpperBound())
return observation return observation
render = _render
reset = _reset
seed = _seed
step = _step

View File

@@ -174,3 +174,8 @@ class RacecarGymEnv(gym.Env):
reward = -closestPoints[0][8] reward = -closestPoints[0][8]
#print(reward) #print(reward)
return reward return reward
render = _render
reset = _reset
seed = _seed
step = _step

View File

@@ -189,3 +189,8 @@ class RacecarZEDGymEnv(gym.Env):
reward = -closestPoints[0][8] reward = -closestPoints[0][8]
#print(reward) #print(reward)
return reward return reward
render = _render
reset = _reset
seed = _seed
step = _step

View File

@@ -109,3 +109,8 @@ class SimpleHumanoidGymEnv(gym.Env):
reward=self._humanoid.distance reward=self._humanoid.distance
print(reward) print(reward)
return reward return reward
render = _render
reset = _reset
seed = _seed
step = _step

View File

@@ -100,6 +100,7 @@ class MJCFBaseBulletEnv(gym.Env):
rgb_array = rgb_array[:, :, :3] rgb_array = rgb_array[:, :, :3]
return rgb_array return rgb_array
def _close(self): def _close(self):
if (self.ownsPhysicsClient): if (self.ownsPhysicsClient):
if (self.physicsClientId>=0): if (self.physicsClientId>=0):
@@ -109,6 +110,17 @@ class MJCFBaseBulletEnv(gym.Env):
def HUD(self, state, a, done): def HUD(self, state, a, done):
pass pass
# backwards compatibility for gym >= v0.9.x
# for extension of this class.
def step(self, *args, **kwargs):
return self._step(*args, **kwargs)
close = _close
render = _render
reset = _reset
seed = _seed
class Camera: class Camera:
def __init__(self): def __init__(self):
pass pass