Merge branch 'main' of github.com:xlang-ai/OSWorld

2025-07-20 07:55:57 +00:00
parent 88d5639a2a e10dd9267c
commit 91bc6bb6ce
6 changed files with 430 additions and 254 deletions
--- a/lib_run_single.py
+++ b/lib_run_single.py
@@ -177,9 +177,7 @@ def run_single_example_opencua(agent, env, example, max_steps, instruction, args
            action_timestamp = datetime.datetime.now().strftime("%Y%m%d@%H%M%S")
            logger.info("Step %d: %s", step_idx + 1, action)
            
-            obs, reward, done, info = env.step(action)
-            time.sleep(3)
-            obs = env._get_obs()
+            obs, reward, done, info = env.step(action, args.sleep_after_execution)

            logger.info(f"Action {action} executed, reward: {reward}, done: {done}")
            # Save screenshot and trajectory information