7fe463b5dd
When episodes are removed from a LeRobotDataset, the remaining episode indices are no longer sequential, which causes indexing errors in get_episode_data(). This happens because episode_data_index tensors are always indexed sequentially, while the episode indices can be arbitrary. This commit introduces a helper function to make the conversion. |
||
|---|---|---|
| .. | ||
| configure_motor.py | ||
| control_robot.py | ||
| control_sim_robot.py | ||
| display_sys_info.py | ||
| eval.py | ||
| find_motors_bus_port.py | ||
| push_dataset_to_hub.py | ||
| push_pretrained.py | ||
| remove_episodes.py | ||
| train.py | ||
| visualize_dataset.py | ||
| visualize_dataset_html.py | ||
| visualize_image_transforms.py | ||