This PR adds the root argument to the dataset visualizer. This parameter is passed to LeRobotDataset.__init__, which enables visualizing local datasets, as an alternative to setting the DATA_DIR env variable.
closes #248
How it was tested
Explain/show how you tested your changes.
added a unit test to tests/test_visualize_dataset that visualizes the local copy of the lerobot/pusht dataset
What this does
This PR adds the
root
argument to the dataset visualizer. This parameter is passed toLeRobotDataset.__init__
, which enables visualizing local datasets, as an alternative to setting theDATA_DIR
env variable.closes #248
How it was tested
Explain/show how you tested your changes.
tests/test_visualize_dataset
that visualizes the local copy of thelerobot/pusht
datasetHow to checkout & try? (for the reviewer)
Examples:
This change is