Skip to content

Commit 9e2001a

Browse files
WIP:
1 parent 6d534c9 commit 9e2001a

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

crisp_gym/record/recording_manager.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -304,6 +304,10 @@ def record_episode(
304304
time.sleep(sleep_time)
305305
continue
306306

307+
logger.debug(
308+
f"Data function returned obs keys: action: {action} for task: {obs['task']}"
309+
)
310+
307311
self.queue.put({"type": "FRAME", "data": (obs, action, task)})
308312

309313
sleep_time = 1 / self.config.fps - (time.time() - frame_start)
@@ -507,7 +511,7 @@ def _on_press(self, key: keyboard.KeyCode | keyboard.Key | None) -> None:
507511
self.state = "exit"
508512
if key.char == "s":
509513
self.state = "to_be_saved"
510-
if key.char == "d":
514+
if key.char == "x":
511515
self.state = "to_be_deleted"
512516
except AttributeError:
513517
pass

0 commit comments

Comments
 (0)