fix runtime tests

All-Hands-AI · Feb 6, 2025 · 46008f7 · 46008f7
1 parent 92aa08e
commit 46008f7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/runtime/test_env_vars.py b/tests/runtime/test_env_vars.py
@@ -86,7 +86,7 @@ def test_env_vars_added_by_config(temp_dir, runtime_cls):
 def test_docker_runtime_env_vars_persist_after_restart(temp_dir):
     from openhands.runtime.impl.docker.docker_runtime import DockerRuntime
 
-    runtime = _load_runtime(temp_dir, DockerRuntime)
+    runtime, config = _load_runtime(temp_dir, DockerRuntime)
 
     # Add a test environment variable
     runtime.add_env_vars({'GITHUB_TOKEN': 'test_token'})

diff --git a/tests/runtime/test_replay.py b/tests/runtime/test_replay.py
@@ -58,7 +58,7 @@ def test_simple_gui_replay(temp_dir, runtime_cls, run_as_openhands):
     2. In GUI mode, agents typically don't finish; rather, they wait for the next
     task from the user, so this exported trajectory ends with awaiting_user_input
     """
-    runtime = _load_runtime(temp_dir, runtime_cls, run_as_openhands)
+    runtime, config = _load_runtime(temp_dir, runtime_cls, run_as_openhands)
 
     config = _get_config('basic_gui_mode')
 
@@ -118,7 +118,7 @@ def test_replay_basic_interactions(temp_dir, runtime_cls, run_as_openhands):
     interference (no asking for user input).
     2) The user messages in the trajectory should appear in the history.
     """
-    runtime = _load_runtime(temp_dir, runtime_cls, run_as_openhands)
+    runtime, config = _load_runtime(temp_dir, runtime_cls, run_as_openhands)
 
     config = _get_config('basic_interactions')