From ec4bf115d0123f7836a5aa92c02e44353947d96c Mon Sep 17 00:00:00 2001 From: Cadene Date: Wed, 10 Apr 2024 13:55:59 +0000 Subject: [PATCH] fix example_1 test --- examples/1_visualize_dataset.py | 2 +- tests/test_examples.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/1_visualize_dataset.py b/examples/1_visualize_dataset.py index a9406d1e..bfc2635f 100644 --- a/examples/1_visualize_dataset.py +++ b/examples/1_visualize_dataset.py @@ -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'] diff --git a/tests/test_examples.py b/tests/test_examples.py index 4263e452..17f6d600 100644 --- a/tests/test_examples.py +++ b/tests/test_examples.py @@ -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():