Update gym_env.py

wrapper_args is a list of lists, not list of dicts
MushroomRL · Jun 25, 2024 · a7d888f · a7d888f
1 parent 752e809
commit a7d888f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mushroom_rl/environments/gym_env.py b/mushroom_rl/environments/gym_env.py
@@ -53,7 +53,7 @@ def __init__(self, name, horizon=None, gamma=0.99, wrappers=None, wrappers_args=
 
         if wrappers is not None:
             if wrappers_args is None:
-                wrappers_args = [dict()] * len(wrappers)
+                wrappers_args = [list()] * len(wrappers)
             for wrapper, args in zip(wrappers, wrappers_args):
                 if isinstance(wrapper, tuple):
                     self.env = wrapper[0](self.env, *args, **wrapper[1])