Fixing linting

HumanCompatibleAI · Feb 1, 2025 · 06090f2 · 06090f2
1 parent a297040
commit 06090f2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/human_aware_rl/rllib/rllib.py b/src/human_aware_rl/rllib/rllib.py
@@ -218,7 +218,9 @@ def _validate_schedule(self, schedule):
     def _setup_action_space(self, agents):
         action_sp = {}
         for agent in agents:
-            action_sp[agent] = gymnasium.spaces.Discrete(len(Action.ALL_ACTIONS))
+            action_sp[agent] = gymnasium.spaces.Discrete(
+                len(Action.ALL_ACTIONS)
+            )
         self.action_space = gymnasium.spaces.Dict(action_sp)
         self.shared_action_space = gymnasium.spaces.Discrete(
             len(Action.ALL_ACTIONS)