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):

    def get_context_mask(self):
        return np.hstack([