From eaf7b9e48fb7bc394d49990bc8fdc62bea055ebd Mon Sep 17 00:00:00 2001 From: yuanmengqi Date: Sat, 7 Jun 2025 21:17:18 +0000 Subject: [PATCH] refactor: replace hardcoded AMI ID with dynamic retrieval from IMAGE_ID_MAP in AWS DesktopEnv initialization --- run_multienv_openaicua.py | 6 ++++-- run_test_env.py | 6 ++++-- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/run_multienv_openaicua.py b/run_multienv_openaicua.py index 65464d0..82af5c7 100644 --- a/run_multienv_openaicua.py +++ b/run_multienv_openaicua.py @@ -176,13 +176,15 @@ def run_env_tasks(env_idx: int, env_tasks: dict, args: argparse.Namespace, share signal.signal(signal.SIGTERM, lambda signum, frame: process_signal_handler(signum, frame, env_idx)) # ami-05e7d7bd279ea4f14 + from desktop_env.providers.aws.manager import IMAGE_ID_MAP + REGION = "us-east-1" env = DesktopEnv( path_to_vm=args.path_to_vm, action_space=args.action_space, provider_name="aws", - region="us-east-1", - snapshot_name="ami-02fea2e5b77c79c17", + region=REGION, + snapshot_name=IMAGE_ID_MAP[REGION], screen_size=(args.screen_width, args.screen_height), headless=args.headless, diff --git a/run_test_env.py b/run_test_env.py index feba441..b4b85f2 100644 --- a/run_test_env.py +++ b/run_test_env.py @@ -223,13 +223,15 @@ def test(args: argparse.Namespace, test_all_meta: dict) -> None: ) agents.append(agent) + from desktop_env.providers.aws.manager import IMAGE_ID_MAP + REGION = "us-east-1" env = DesktopEnv( path_to_vm=args.path_to_vm, action_space=agent.action_space, provider_name="aws", - region="us-east-1", - snapshot_name="ami-05e7d7bd279ea4f14", + region=REGION, + snapshot_name=IMAGE_ID_MAP[REGION], screen_size=(args.screen_width, args.screen_height), headless=args.headless,