From 4b9a81080b3daa9df30e9fea27d07e248b6b8b3c Mon Sep 17 00:00:00 2001 From: Johannes Fischer Date: Fri, 8 Oct 2021 18:17:02 +0200 Subject: [PATCH] Add missing * in **kwargs --- scratch/etienne/intersimple/gail_options_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scratch/etienne/intersimple/gail_options_image.py b/scratch/etienne/intersimple/gail_options_image.py index ceddd17..ae70490 100644 --- 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({