def spaces(self):
comb_spaces = [
EnvSpaces(observation=self.player_observation_space,
action=self.player_action_space),
EnvSpaces(observation=self.observer_observation_space,
action=self.observer_action_space)
]
return comb_spaces
def __init__(self, wrapped_env, act_null_value=0, force_float32=True):
self._wrapped_env = wrapped_env
action_dim = 3
self.action_space = GymSpaceWrapper(
space=gym.spaces.Box(low=-1,
high=1,
shape=(action_dim, ),
dtype=np.float32),
name="act",
null_value=act_null_value,
force_float32=force_float32,
)
self.observation_space = Composite([
Box(low=-np.inf, high=np.inf, shape=(64, 64, 3), dtype=np.float32),
Box(np.array([-1] * 100), np.array([1] * 100), dtype=np.float32)
], OBS)
self.spaces = EnvSpaces(observation=self.observation_space,
action=self.action_space)
self._dtype = None
self.current_location = None
def spaces(self):
"""Returns the rlpyt spaces for the wrapped env."""
return EnvSpaces(
observation=self.observation_space,
action=self.action_space,
)
def spaces(self):
return EnvSpaces(
observation=self._observation_space,
action=self._action_space,
)