def get_state_space(self): if self.processor is None: return Space.convert_openai_space(self.env.observation_space) else: self.reset() shape = self.current_state.shape min = np.zeros(shape, dtype=np.uint8) max = np.full(shape, 255, dtype=np.uint8) return Space(min, max, True)
def get_action_space(self): return Space.convert_openai_space(self.env.action_space)