diff --git a/lib_run_single.py b/lib_run_single.py index c2742df..edc1900 100644 --- a/lib_run_single.py +++ b/lib_run_single.py @@ -2,7 +2,7 @@ import datetime import json import logging import os - +import time from wrapt_timeout_decorator import * 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): runtime_logger = setup_logger(example, example_result_dir) 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 step_idx = 0 env.controller.start_recording()