Add missing * in **kwargs

2021-10-08 18:17:02 +02:00
parent 59681cb16f
commit 4b9a81080b
1 changed files with 1 additions and 1 deletions
--- a/scratch/etienne/intersimple/gail_options_image.py
+++ b/scratch/etienne/intersimple/gail_options_image.py
@@ -48,7 +48,7 @@ class OptionsCnnPolicy(stable_baselines3.common.policies.ActorCriticCnnPolicy):
 class OptionsEnv(gym.Wrapper):
    
    def __init__(self, env, *args, **kwargs):
-        super().__init__(env, *args, *kwargs)
+        super().__init__(env, *args, **kwargs)
        num_hl_options = len(ALL_OPTIONS)
        self.action_space = gym.spaces.Discrete(num_hl_options)
        self.observation_space = gym.spaces.Dict({