fix: remove unnecessary sleep and observation retrieval in run_single_example function

This commit is contained in:
yuanmengqi
2025-07-25 15:51:20 +00:00
parent f5595df71c
commit 39e5baf5ae
2 changed files with 1 additions and 2 deletions

View File

@@ -32,8 +32,6 @@ def run_single_example(agent, env, example, max_steps, instruction, args, exampl
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, args.sleep_after_execution)
time.sleep(3)
obs = env._get_obs()
logger.info("Reward: %.2f", reward)
logger.info("Done: %s", done)

View File

@@ -63,3 +63,4 @@ docker
loguru
dotenv
tldextract
anthropic