lzy/sci-gui-agent-benchmark
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: replace hardcoded AMI ID with dynamic retrieval from IMAGE_ID_MAP in AWS DesktopEnv initialization

Browse Source
This commit is contained in:
yuanmengqi
2025-06-07 21:17:18 +00:00
parent 8853671220
commit eaf7b9e48f
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
run_multienv_openaicua.py
View File

@@ -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)) signal.signal(signal.SIGTERM, lambda signum, frame: process_signal_handler(signum, frame, env_idx))
# ami-05e7d7bd279ea4f14 # ami-05e7d7bd279ea4f14
from desktop_env.providers.aws.manager import IMAGE_ID_MAP
REGION = "us-east-1"
env = DesktopEnv( env = DesktopEnv(
path_to_vm=args.path_to_vm, path_to_vm=args.path_to_vm,
action_space=args.action_space, action_space=args.action_space,
provider_name="aws", provider_name="aws",
region="us-east-1", region=REGION,
snapshot_name="ami-02fea2e5b77c79c17", snapshot_name=IMAGE_ID_MAP[REGION],
screen_size=(args.screen_width, args.screen_height), screen_size=(args.screen_width, args.screen_height),
headless=args.headless, headless=args.headless,

6
run_test_env.py
View File

@@ -223,13 +223,15 @@ def test(args: argparse.Namespace, test_all_meta: dict) -> None:
) )
agents.append(agent) agents.append(agent)
from desktop_env.providers.aws.manager import IMAGE_ID_MAP
REGION = "us-east-1"
env = DesktopEnv( env = DesktopEnv(
path_to_vm=args.path_to_vm, path_to_vm=args.path_to_vm,
action_space=agent.action_space, action_space=agent.action_space,
provider_name="aws", provider_name="aws",
region="us-east-1", region=REGION,
snapshot_name="ami-05e7d7bd279ea4f14", snapshot_name=IMAGE_ID_MAP[REGION],
screen_size=(args.screen_width, args.screen_height), screen_size=(args.screen_width, args.screen_height),
headless=args.headless, headless=args.headless,