lzy/sci-gui-agent-benchmark
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Modify: wait for initial env to be ready (#139)

Browse Source
This commit is contained in:
Xinyuan Wang
2025-02-25 21:08:25 +08:00
committed by GitHub
parent eb9758774f
commit d626cc90d9
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib_run_single.py
View File

@@ -2,7 +2,7 @@ import datetime
import json import json
import logging import logging
import os import os
import time
from wrapt_timeout_decorator import * from wrapt_timeout_decorator import *
logger = logging.getLogger("desktopenv.experiment") logger = logging.getLogger("desktopenv.experiment")
@@ -11,7 +11,9 @@ logger = logging.getLogger("desktopenv.experiment")
def run_single_example(agent, env, example, max_steps, instruction, args, example_result_dir, scores): def run_single_example(agent, env, example, max_steps, instruction, args, example_result_dir, scores):
runtime_logger = setup_logger(example, example_result_dir) runtime_logger = setup_logger(example, example_result_dir)
agent.reset(runtime_logger) agent.reset(runtime_logger)
obs = env.reset(task_config=example) env.reset(task_config=example)
time.sleep(60) # Wait for the environment to be ready
obs = env._get_obs() # Get the initial observation
done = False done = False
step_idx = 0 step_idx = 0
env.controller.start_recording() env.controller.start_recording()