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

fix example_1 test

This commit is contained in:
Cadene 2024-04-10 13:55:59 +00:00
parent c08003278e
commit ec4bf115d0
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/1_visualize_dataset.py
View File

@ -17,4 +17,4 @@ video_paths = render_dataset(
max_num_episodes=1,
)
print(video_paths)
# ['outputs/visualize_dataset/example/episode_0.mp4']
# ['outputs/visualize_dataset/example/episode_0_top.mp4']

2
tests/test_examples.py
View File

@ -15,7 +15,7 @@ def test_example_1():
file_contents = file.read()
exec(file_contents)
assert Path("outputs/visualize_dataset/example/episode_0.mp4").exists()
assert Path("outputs/visualize_dataset/example/episode_0_top.mp4").exists()
def test_examples_3_and_2():