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

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,

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,