robosuite
robosuite copied to clipboard
Modify DataCollectionWrapper and collect_human_demonstrations script to save only successful demonstrations
Modify DataCollectionWrapper and collect_human_demonstrations script to save only successful demonstrations to dataset, and disregard any unsuccessful demonstrations.
LGTM - as the last step, please check our contributing guidelines and run the pre-commit checks. The updated robosuite/wrappers/data_collection_wrapper.py code does not comply with the black formatter and the pre-commit hooks will correct the style automatically.
This has been merged - thanks for your contribution @PraveenElango