aws_communication_success

run_multienv.py

@@ -107,6 +107,11 @@ def config() -> argparse.Namespace:
     # logging related
     parser.add_argument("--result_dir", type=str, default="./results")
     parser.add_argument("--num_envs", type=int, default=1, help="Number of environments to run in parallel")
+
+    # aws config
+    parser.add_argument(
+        "--region", type=str, default="us-east-1", help="AWS region for the VM"
+    )
     
     args = parser.parse_args()
     return args
@@ -220,6 +225,7 @@ def test(args: argparse.Namespace, test_all_meta: dict) -> None:
 
         env = DesktopEnv(
             path_to_vm=args.path_to_vm,
+            region=args.region,
             action_space=agent.action_space,
             screen_size=(args.screen_width, args.screen_height),
             headless=args.headless,
@@ -330,6 +336,7 @@ def get_result(action_space, use_model, observation_type, result_dir, total_file
 if __name__ == "__main__":
     ####### The complete version of the list of examples #######
     os.environ["TOKENIZERS_PARALLELISM"] = "false"
+
     args = config()
 
     with open(args.test_all_meta_path, "r", encoding="utf-8") as f:
@@ -358,3 +365,6 @@ if __name__ == "__main__":
         test_all_meta,
     )
     test(args, test_file_list)
+
+
+# path_to_vm can be a list["xxx","xxx"]