Exclude interactive example from running during tests

HumanCompatibleAI · Sep 7, 2023 · f6ebbc2 · f6ebbc2
1 parent 79749dd
commit f6ebbc2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -30,9 +30,12 @@ def _paths_to_strs(x: Iterable[pathlib.Path]) -> Sequence[str]:
 EXAMPLES_DIR = THIS_DIR / ".." / "examples"
 TUTORIALS_DIR = THIS_DIR / ".." / "docs" / "tutorials"
 
-SH_PATHS = _paths_to_strs(EXAMPLES_DIR.glob("*.sh"))
+EXCLUDED_EXAMPLE_FILES = ["train_dagger_atari_interactive_policy.py"]
+EXCLUDED_EXAMPLE_PATHS = [THIS_DIR / ".." / "examples" / f for f in EXCLUDED_EXAMPLE_FILES]
+
+SH_PATHS = _paths_to_strs(set(EXAMPLES_DIR.glob("*.sh")) - set(EXCLUDED_EXAMPLE_PATHS))
 TUTORIAL_PATHS = _paths_to_strs(TUTORIALS_DIR.glob("*.ipynb"))
-PY_PATHS = _paths_to_strs(EXAMPLES_DIR.glob("*.py"))
+PY_PATHS = _paths_to_strs(set(EXAMPLES_DIR.glob("*.py")) - set(EXCLUDED_EXAMPLE_PATHS))
 
 
 # Note: This is excluded from coverage since is computed on linux. However, it is