add ProDMP to register

2022-10-13 10:57:00 +02:00
parent d73c9bbdbf
commit 187c5f5bb2
5 changed files with 41 additions and 5 deletions
@@ -5,7 +5,7 @@ from gym import register
 from . import mujoco
 from .deprecated_needs_gym_robotics import robotics

-ALL_GYM_MOVEMENT_PRIMITIVE_ENVIRONMENTS = {"DMP": [], "ProMP": []}
+ALL_GYM_MOVEMENT_PRIMITIVE_ENVIRONMENTS = {"DMP": [], "ProMP": [], "ProDMP": []}

 DEFAULT_BB_DICT_ProMP = {
    "name": 'EnvName',