Merge branch 'zdy'
This commit is contained in:
19
main.py
19
main.py
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user