Merge branch 'zdy'

2023-12-22 15:23:56 +08:00
parent aa76a57437 fffc4aadca
commit 4782b687b9
2 changed files with 43 additions and 46 deletions
--- a/main.py
+++ b/main.py
@@ -9,14 +9,11 @@ def human_agent():

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

    env = DesktopEnv( path_to_vm=r"C:\Users\tianbaox\Documents\Virtual Machines\Ubuntu\Ubuntu.vmx"
                    , action_space="computer_13"
-                    , snapshot_path="base_setup"
-                    , task_id=example["id"]
-                    , instruction=example["instruction"]
-                    , config=example["config"]
-                    , evaluator=example["evaluator"]
+                    , task_config=example
                    )

    # reset the environment to certain snapshot