diff --git a/fancy_gym/envs/mujoco/reacher/mp_wrapper.py b/fancy_gym/envs/mujoco/reacher/mp_wrapper.py index 0464640..d47737a 100644 --- a/fancy_gym/envs/mujoco/reacher/mp_wrapper.py +++ b/fancy_gym/envs/mujoco/reacher/mp_wrapper.py @@ -7,6 +7,16 @@ from fancy_gym.black_box.raw_interface_wrapper import RawInterfaceWrapper class MPWrapper(RawInterfaceWrapper): + mp_config = { + 'ProMP': {}, + 'DMP': { + 'phase_generator_kwargs': { + 'alpha_phase': 2, + }, + }, + 'ProDMP': {}, + } + @property def context_mask(self): return np.concatenate([[False] * self.n_links, # cos