huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add missing * in **kwargs

Browse Source
This commit is contained in:
Johannes Fischer
2021-10-08 18:17:02 +02:00
parent 59681cb16f
commit 4b9a81080b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scratch/etienne/intersimple/gail_options_image.py
View File

@@ -48,7 +48,7 @@ class OptionsCnnPolicy(stable_baselines3.common.policies.ActorCriticCnnPolicy):
class OptionsEnv(gym.Wrapper): class OptionsEnv(gym.Wrapper):
def __init__(self, env, *args, **kwargs): def __init__(self, env, *args, **kwargs):
super().__init__(env, *args, *kwargs) super().__init__(env, *args, **kwargs)
num_hl_options = len(ALL_OPTIONS) num_hl_options = len(ALL_OPTIONS)
self.action_space = gym.spaces.Discrete(num_hl_options) self.action_space = gym.spaces.Discrete(num_hl_options)
self.observation_space = gym.spaces.Dict({ self.observation_space = gym.spaces.Dict({