admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 09:46:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

pre-commit run

Browse Source
This commit is contained in:
Steven Palma
2026-04-14 15:42:19 +02:00
parent 05a2604d6e
commit f55782f9f7
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/rollout/strategies/dagger.py
+5 -1
View File
@@ -151,7 +151,11 @@ def _init_dagger_keyboard():
def on_press(key): def on_press(key):
try: try:
if events["in_reset"]: if events["in_reset"]:
if key in [keyboard.Key.space, keyboard.Key.right] or hasattr(key, "char") and key.char == "c": if (
key in [keyboard.Key.space, keyboard.Key.right]
or hasattr(key, "char")
and key.char == "c"
):
events["start_next_episode"] = True events["start_next_episode"] = True
elif key == keyboard.Key.esc: elif key == keyboard.Key.esc:
events["stop_recording"] = True events["stop_recording"] = True