fix: remove unnecessary sleep and observation retrieval in run_single_example function
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -63,3 +63,4 @@ docker
|
||||
loguru
|
||||
dotenv
|
||||
tldextract
|
||||
anthropic
|
||||
|
||||
Reference in New Issue
Block a user