default observation selection for MPs

This commit is contained in:
ottofabian 2021-05-18 10:39:30 +02:00
parent 528b7521b6
commit 7695cae076

View File

@ -10,17 +10,17 @@ class MPEnv(gym.Env):
@property @property
@abstractmethod @abstractmethod
def active_obs(self): def active_obs(self):
"""Returns boolean value for each observation entry """Returns boolean mask for each observation entry
whether the observation is returned by the DMP for the contextual case or not. whether the observation is returned for the contextual case or not.
This effectively allows to filter unwanted or unnecessary observations from the full step-based case. This effectively allows to filter unwanted or unnecessary observations from the full step-based case.
""" """
raise NotImplementedError() return np.ones(self.observation_space.shape, dtype=bool)
@property @property
@abstractmethod @abstractmethod
def start_pos(self) -> Union[float, int, np.ndarray]: def start_pos(self) -> Union[float, int, np.ndarray]:
""" """
Returns the current position of the joints Returns the starting position of the joints
""" """
raise NotImplementedError() raise NotImplementedError()