Setup initial examples

This commit is contained in:
Timothyxxx
2023-12-06 17:04:33 +08:00
parent 4ba053998d
commit b9c317f0f5
14 changed files with 285 additions and 6 deletions

12
main.py
View File

@@ -1,3 +1,4 @@
import json
from desktop_env.envs.desktop_env import DesktopEnv
@@ -5,17 +6,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:
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\Win10\Win10.vmx""",
# path_to_vm="/home/yuri/vmware/Ubuntu 64-bit/Ubuntu 64-bit.vmx",
snapshot_path="base_setup",
config={
"download": [(
"https://drive.usercontent.google.com/download?id=1rwhniaClEkF8XFzdfaNUA6GmAiy4syMZ&export=download&authuser=0&confirm=t&uuid=6fdd5b04-85f4-45e1-ad74-368f8f2a82ab&at=APZUnTUP-JxPxLfNls6jXWghblQ5:1701766091851",
r"C:\Users\tianbaox\Desktop\Quarterly_Product_Sales_by_Zone.xlsx")],
"open": [r"C:\Users\tianbaox\Desktop\Quarterly_Product_Sales_by_Zone.xlsx"],
}
config=example["config"],
)
# reset the environment to certain snapshot