ver Dec22ndv3

added function to switch tasks and reload the configs of setup & evaluation
2023-12-22 15:20:27 +08:00
parent 10c8fbe995
commit fffc4aadca
2 changed files with 43 additions and 47 deletions
--- a/main.py
+++ b/main.py
@@ -9,17 +9,13 @@ def human_agent():

    with open("evaluation_examples/examples/f9584479-3d0d-4c79-affa-9ad7afdd8850.json", "r") as f:
        example = json.load(f)
+    example["snapshot"] = "Init6"

    #env = DesktopEnv( path_to_vm="/home/yuri/vmware/Windows 10 x64/Windows 10 x64.vmx"
                    #  path_to_vm="/home/yuri/vmware/Ubuntu 64-bit/Ubuntu 64-bit.vmx",
    env = DesktopEnv( path_to_vm="/home/david/vmware/KUbuntu 64-bit/KUbuntu 64-bit.vmx"
                    , action_space="computer_13"
-                     #, snapshot_path="base_setup"
-                    , snapshot_path="Init6"
-                    , task_id=example["id"]
-                    , instruction=example["instruction"]
-                    , config=example["config"]
-                    , evaluator=example["evaluator"]
+                    , task_config=example
                    )

    # reset the environment to certain snapshot