fixed OpenAI fetch tasks; added nicer imports

alr_envs/dmc/__init__.py

@@ -0,0 +1,5 @@
+# from alr_envs.dmc import manipulation, suite
+from alr_envs.dmc.suite import ball_in_cup
+from alr_envs.dmc.suite import reacher
+from alr_envs.dmc.suite import cartpole
+from alr_envs.dmc.manipulation import reach

alr_envs/dmc/manipulation/reach/__init__.py

@@ -0,0 +1 @@
+from .mp_wrapper import MPWrapper

alr_envs/dmc/manipulation/reach/mp_wrapper.py

@@ -2,10 +2,10 @@ from typing import Tuple, Union
 
 import numpy as np
 
-from mp_env_api.interface_wrappers.mp_env_wrapper import MPEnvWrapper
+from mp_env_api import MPEnvWrapper
 
 
-class DMCReachSiteMPWrapper(MPEnvWrapper):
+class MPWrapper(MPEnvWrapper):
 
     @property
     def active_obs(self):

alr_envs/dmc/suite/ball_in_cup/__init__.py

@@ -0,0 +1 @@
+from .mp_wrapper import MPWrapper

alr_envs/dmc/suite/ball_in_cup/mp_wrapper.py

@@ -2,10 +2,10 @@ from typing import Tuple, Union
 
 import numpy as np
 
-from mp_env_api.interface_wrappers.mp_env_wrapper import MPEnvWrapper
+from mp_env_api import MPEnvWrapper
 
 
-class DMCBallInCupMPWrapper(MPEnvWrapper):
+class MPWrapper(MPEnvWrapper):
 
     @property
     def active_obs(self):

alr_envs/dmc/suite/cartpole/__init__.py

@@ -0,0 +1,3 @@
+from .mp_wrapper import MPWrapper
+from .mp_wrapper import TwoPolesMPWrapper
+from .mp_wrapper import ThreePolesMPWrapper

alr_envs/dmc/suite/cartpole/mp_wrapper.py

@@ -2,10 +2,10 @@ from typing import Tuple, Union
 
 import numpy as np
 
-from mp_env_api.interface_wrappers.mp_env_wrapper import MPEnvWrapper
+from mp_env_api import MPEnvWrapper
 
 
-class DMCCartpoleMPWrapper(MPEnvWrapper):
+class MPWrapper(MPEnvWrapper):
 
     def __init__(self, env, n_poles: int = 1):
         self.n_poles = n_poles
@@ -39,13 +39,13 @@ class DMCCartpoleMPWrapper(MPEnvWrapper):
         return self.env.dt
 
 
-class DMCCartpoleTwoPolesMPWrapper(DMCCartpoleMPWrapper):
+class TwoPolesMPWrapper(MPWrapper):
 
     def __init__(self, env):
         super().__init__(env, n_poles=2)
 
 
-class DMCCartpoleThreePolesMPWrapper(DMCCartpoleMPWrapper):
+class ThreePolesMPWrapper(MPWrapper):
 
     def __init__(self, env):
         super().__init__(env, n_poles=3)

alr_envs/dmc/suite/reacher/__init__.py

@@ -0,0 +1 @@
+from .mp_wrapper import MPWrapper

alr_envs/dmc/suite/reacher/mp_wrapper.py

@@ -2,10 +2,10 @@ from typing import Tuple, Union
 
 import numpy as np
 
-from mp_env_api.interface_wrappers.mp_env_wrapper import MPEnvWrapper
+from mp_env_api import MPEnvWrapper
 
 
-class DMCReacherMPWrapper(MPEnvWrapper):
+class MPWrapper(MPEnvWrapper):
 
     @property
     def active_obs(self):