Open ChorntonYoel opened 6 days ago
Also style check failing was failing. Could you please run our pre-commit? :D
pre-commit install
pre-commit run --all-files
https://github.com/huggingface/lerobot/blob/main/CONTRIBUTING.md#submitting-a-pull-request-pr
What this does
Those PR is meant to add the possibility for the teleoperator to add rewards when performing the tasks. The current approach simply consists in having frames labeled 0 until the experimenter presses the space bar. Then the reward for each frame becomes 1. The experimenter can press the bar again (eg. in case of subsequent failure) and the frame labeling will return to 0. Each time the space bar is pressed, the labeling switches
This is one way to go about it, but lmk if you have a better idea!
How it was tested
I made datasets with Moss with and without rewards. Checking that I got the expected behavior.
How to checkout & try? (for the reviewer)