[FIX] wrapper_args is a list of dicts

MushroomRL · Sep 2, 2024 · 72070fc · 72070fc
1 parent c6ce144
commit 72070fc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/mushroom_rl/environments/gym_env.py b/mushroom_rl/environments/gym_env.py
@@ -36,7 +36,7 @@ def __init__(self, name, horizon=None, gamma=0.99, wrappers=None, wrappers_args=
                 a tuple with two elements: the gym wrapper class and a
                 dictionary containing the parameters needed by the wrapper
                 constructor;
-            wrappers_args (list, None): list of list of arguments for each wrapper;
+            wrappers_args (list, None): list of dictionaries of arguments for each wrapper;
             ** env_args: other gym environment parameters.
 
         """
@@ -56,9 +56,9 @@ def __init__(self, name, horizon=None, gamma=0.99, wrappers=None, wrappers_args=
                 wrappers_args = [dict()] * len(wrappers)
             for wrapper, args in zip(wrappers, wrappers_args):
                 if isinstance(wrapper, tuple):
-                    self.env = wrapper[0](self.env, *args, **wrapper[1])
+                    self.env = wrapper[0](self.env, **args, **wrapper[1])
                 else:
-                    self.env = wrapper(self.env, *args, **env_args)
+                    self.env = wrapper(self.env, **args, **env_args)
 
         horizon = self._set_horizon(self.env, horizon)
 

diff --git a/mushroom_rl/environments/gymnasium_env.py b/mushroom_rl/environments/gymnasium_env.py
@@ -38,7 +38,7 @@ def __init__(self, name, horizon=None, gamma=0.99, headless = False, wrappers=No
                 a tuple with two elements: the gym wrapper class and a
                 dictionary containing the parameters needed by the wrapper
                 constructor;
-            wrappers_args (list, None): list of list of arguments for each wrapper;
+            wrappers_args (list, None): list of dictionaries of arguments for each wrapper;
             ** env_args: other gym environment parameters.
 
         """
@@ -60,9 +60,9 @@ def __init__(self, name, horizon=None, gamma=0.99, headless = False, wrappers=No
                 wrappers_args = [dict()] * len(wrappers)
             for wrapper, args in zip(wrappers, wrappers_args):
                 if isinstance(wrapper, tuple):
-                    self.env = wrapper[0](self.env, *args, **wrapper[1])
+                    self.env = wrapper[0](self.env, **args, **wrapper[1])
                 else:
-                    self.env = wrapper(self.env, *args, **env_args)
+                    self.env = wrapper(self.env, **args, **env_args)
 
         horizon = self._set_horizon(self.env, horizon)