Skip to content

Commit 7e57779

Browse files
committed
update test_run_v1_get
1 parent 5c20f22 commit 7e57779

1 file changed

Lines changed: 3 additions & 6 deletions

File tree

tests/test_api/test_run.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -29,12 +29,9 @@ def _assert_run_shape(run: OpenMLRun) -> None:
2929
assert isinstance(run.task_id, int)
3030

3131

32-
def test_run_v1_get(run_v1, with_test_cache):
33-
import os
34-
35-
# Run 1 exists on the remote test server; the local docker server only seeds run 25.
36-
run_id = 25 if os.getenv("OPENML_USE_LOCAL_SERVICES") == "true" else 1
37-
run = run_v1.get(run_id=run_id)
32+
def test_run_v1_get(run_v1, test_files_directory):
33+
openml.config.set_root_cache_directory(test_files_directory)
34+
run = run_v1.get(run_id=1)
3835
_assert_run_shape(run)
3936

4037

0 commit comments

Comments
 (0)