Merge remote-tracking branch 'upstream/feat/aws-provider-support'

2025-06-08 04:01:35 +00:00
parent d8872634ee eaf7b9e48f
commit 3e541bb393
51 changed files with 139 additions and 125 deletions
--- a/run_multienv_openaicua.py
+++ b/run_multienv_openaicua.py
@@ -181,13 +181,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,