ported mp_config for mujoco/box_pushing

This commit is contained in:
Dominik Moritz Roth 2023-07-20 11:44:50 +02:00
parent 17d370e2ba
commit 8d26cccc35

View File

@ -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