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:
@@ -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]
|
||||||
|
|||||||
Reference in New Issue
Block a user