diff --git a/lib_run_single.py b/lib_run_single.py index eb91c72..91a0163 100644 --- a/lib_run_single.py +++ b/lib_run_single.py @@ -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) diff --git a/requirements.txt b/requirements.txt index 924fffe..0a32219 100644 --- a/requirements.txt +++ b/requirements.txt @@ -63,3 +63,4 @@ docker loguru dotenv tldextract +anthropic