Post-rebase fix BENCHMARK_EPISODES_FOLDER in tox *-ml envs.

flatland-association · Feb 13, 2025 · 48347b6 · 48347b6
1 parent 8883e1c
commit 48347b6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/benchmarks/benchmark_episodes.py b/benchmarks/benchmark_episodes.py
@@ -257,6 +257,7 @@ def run_episode(data_sub_dir: str, ep_id: str):
     ep_id the episode ID
     """
     _dir = os.getenv("BENCHMARK_EPISODES_FOLDER")
+    assert _dir is not None, (DOWNLOAD_INSTRUCTIONS, _dir)
     assert os.path.exists(_dir), (DOWNLOAD_INSTRUCTIONS, _dir)
     data_dir = os.path.join(_dir, data_sub_dir)
 

diff --git a/tox.ini b/tox.ini
@@ -42,6 +42,8 @@ commands =
 
 [testenv:py{310,311,312}-ml]
 platform = linux|linux2|darwin
+set_env =
+    BENCHMARK_EPISODES_FOLDER = {tox_root}/episodes
 deps =
     -r requirements-dev.txt
     -r requirements-ml.txt