pre commit

huggingface · Nov 22, 2024 · e73005e · e73005e
1 parent 6394d76
commit e73005e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lerobot/common/robot_devices/control_utils.py b/lerobot/common/robot_devices/control_utils.py
@@ -152,9 +152,12 @@ def on_press(key):
                 events["stop_recording"] = True
                 events["exit_early"] = True
             elif assign_rewards and key == keyboard.Key.space:
-                print("Space key pressed. Assigning new rewards to the frames. New reward:", not events["reward"])
+                print(
+                    "Space key pressed. Assigning new rewards to the frames. New reward:",
+                    not events["reward"],
+                )
                 events["reward"] = not events["reward"]
-                
+
         except Exception as e:
             print(f"Error handling key press: {e}")