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")
|
action_timestamp = datetime.datetime.now().strftime("%Y%m%d@%H%M%S")
|
||||||
logger.info("Step %d: %s", step_idx + 1, action)
|
logger.info("Step %d: %s", step_idx + 1, action)
|
||||||
obs, reward, done, info = env.step(action, args.sleep_after_execution)
|
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("Reward: %.2f", reward)
|
||||||
logger.info("Done: %s", done)
|
logger.info("Done: %s", done)
|
||||||
|
|||||||
@@ -63,3 +63,4 @@ docker
|
|||||||
loguru
|
loguru
|
||||||
dotenv
|
dotenv
|
||||||
tldextract
|
tldextract
|
||||||
|
anthropic
|
||||||
|
|||||||
Reference in New Issue
Block a user