Merge branch 'zdy'

This commit is contained in:
David Chang
2023-12-21 10:42:18 +08:00
6 changed files with 106 additions and 64 deletions

19
main.py
View File

@@ -7,19 +7,16 @@ def human_agent():
Runs the Gym environment with human input.
"""
with open("evaluation_examples/examples/37608790-6147-45d0-9f20-1137bb35703d.json", "r") as f:
with open("evaluation_examples/examples/f9584479-3d0d-4c79-affa-9ad7afdd8850.json", "r") as f:
example = json.load(f)
env = DesktopEnv(
# path_to_vm=r"""C:\Users\tianbaox\Downloads\Windows 10 x64\Windows 10 x64.vmx""",
path_to_vm=r"""C:\Users\tianbaox\Documents\Virtual Machines\Ubuntu\Ubuntu.vmx""",
# path_to_vm="/home/yuri/vmware/Ubuntu 64-bit/Ubuntu 64-bit.vmx",
action_space="computer_13",
snapshot_path="base_setup",
instruction=example["instruction"],
# config=example["config"],
# evaluator=example["evaluator"]
)
env = DesktopEnv( path_to_vm=r"C:\Users\tianbaox\Documents\Virtual Machines\Ubuntu\Ubuntu.vmx"
, action_space="computer_13"
, snapshot_path="base_setup"
, instruction=example["instruction"]
, config=example["config"]
, evaluator=example["evaluator"]
)
# reset the environment to certain snapshot
observation = env.reset()