formatting

HumanCompatibleAI · Sep 7, 2023 · f63bf5e · f63bf5e
1 parent 3a1d10f
commit f63bf5e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/imitation/policies/interactive.py b/src/imitation/policies/interactive.py
@@ -144,5 +144,9 @@ def __init__(self, env: typing.Union[gym.Env, vec_env.VecEnv], *args, **kwargs):
             [(ATARI_ACTION_NAMES_TO_KEYS[name], name) for name in action_names],
         )
         super().__init__(
-            env.observation_space, env.action_space, action_keys_names, *args, **kwargs,
+            env.observation_space,
+            env.action_space,
+            action_keys_names,
+            *args,
+            **kwargs,
         )