adjust env registries in __init__

2022-06-30 14:55:34 +02:00
parent 3273f455c5
commit f31d85451f
11 changed files with 29 additions and 65 deletions
@@ -1,2 +1 @@
-from .mp_wrapper import MPWrapper
-from .new_mp_wrapper import NewMPWrapper
+from .new_mp_wrapper import MPWrapper
@@ -5,7 +5,7 @@ import numpy as np
 from alr_envs.mp.raw_interface_wrapper import RawInterfaceWrapper


-class NewMPWrapper(RawInterfaceWrapper):
+class MPWrapper(RawInterfaceWrapper):
    @property
    def current_pos(self) -> Union[float, int, np.ndarray, Tuple]:
        return self.env.sim.data.qpos[0:7].copy()