unskip

2024-03-26 10:45:31 +00:00
parent 98534d1a63
commit 9ced0cf1fb
2 changed files with 3 additions and 3 deletions
--- a/tests/test_examples.py
+++ b/tests/test_examples.py
@@ -1,7 +1,6 @@
 import pytest
 from pathlib import Path

-@pytest.mark.skip(reason="For some reason 1_visualize_dataset.py downloads the dataset")
@pytest.mark.parametrize(
    "path",
    [
@@ -16,4 +15,5 @@ def test_example(path):
        file_contents = file.read()
    exec(file_contents)

-    assert Path("outputs/visualize_dataset/example/episode_0.mp4").exists()
+    if path == "examples/1_visualize_dataset.py":
+        assert Path("outputs/visualize_dataset/example/episode_0.mp4").exists()