wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

WIP

This commit is contained in:
Remi Cadene 2024-05-20 15:09:57 +00:00
parent 4a19733dea
commit 885cb9ddeb
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/visualize_dataset.py
View File

@ -295,10 +295,10 @@ def write_episode_data_html(output_dir, file_name, js_fname, ep_index, dataset):
def write_episodes_list_html(output_dir, file_name, ep_indices, ep_html_fnames, dataset): def write_episodes_list_html(output_dir, file_name, ep_indices, ep_html_fnames, dataset):
"""Write an html file containing information related to the dataset and a list of links to """Write an html file containing information related to the dataset and a list of links to
html pages of episodes.""" html pages of episodes."""
soup, body = create_html_page(dataset.hf_dataset.info.dataset_name) soup, body = create_html_page("TODO")
h3 = soup.new_tag("h3") h3 = soup.new_tag("h3")
h3.string = dataset.hf_dataset.info.dataset_name h3.string = "TODO"
body.append(h3) body.append(h3)
ul_info = soup.new_tag("ul") ul_info = soup.new_tag("ul")

5
tests/test_visualize_dataset.py
View File

@ -25,9 +25,8 @@ from lerobot.scripts.visualize_dataset import visualize_dataset
def test_visualize_dataset(tmpdir, repo_id): def test_visualize_dataset(tmpdir, repo_id):
rrd_path = visualize_dataset( rrd_path = visualize_dataset(
repo_id, repo_id,
episode_index=0, episode_indices=[0],
batch_size=32,
save=True,
output_dir=tmpdir, output_dir=tmpdir,
serve=False,
) )
assert rrd_path.exists() assert rrd_path.exists()