1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00

removing some Gym wrapper code that was required for OpenAI Universe in the past

This commit is contained in:
Itai Caspi
2017-10-25 10:01:03 +03:00
parent eb0b57d7fa
commit d3c6860421

View File

@@ -46,15 +46,13 @@ i = 0
class GymEnvironmentWrapper(EnvironmentWrapper): class GymEnvironmentWrapper(EnvironmentWrapper):
def __init__(self, tuning_parameters): def __init__(self, tuning_parameters):
EnvironmentWrapper.__init__(self, tuning_parameters) EnvironmentWrapper.__init__(self, tuning_parameters)
ports = (5200, 15200)
# env parameters # env parameters
self.env = gym.make(self.env_id) self.env = gym.make(self.env_id)
self.env_id = self.env_id
if self.seed is not None: if self.seed is not None:
self.env.seed(self.seed) self.env.seed(self.seed)
self.env_spec = gym.spec(self.env_id) # self.env_spec = gym.spec(self.env_id)
self.none_counter = 0
self.discrete_controls = type(self.env.action_space) != gym.spaces.box.Box self.discrete_controls = type(self.env.action_space) != gym.spaces.box.Box
# pybullet requires rendering before resetting the environment, but other gym environments (Pendulum) will crash # pybullet requires rendering before resetting the environment, but other gym environments (Pendulum) will crash
@@ -70,7 +68,6 @@ class GymEnvironmentWrapper(EnvironmentWrapper):
if self.is_rendered: if self.is_rendered:
self.render() self.render()
# self.env.render()
self.is_state_type_image = len(o.shape) > 1 self.is_state_type_image = len(o.shape) > 1
if self.is_state_type_image: if self.is_state_type_image:
self.width = o.shape[1] self.width = o.shape[1]