lzy/sci-gui-agent-benchmark
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update todos

Browse Source
This commit is contained in:
Timothyxxx
2024-03-14 22:36:33 +08:00
parent 44ff027801
commit 35ed7cec89
2 changed files with 14 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
desktop_env/envs/desktop_env.py
View File

@@ -53,7 +53,7 @@ class DesktopEnv(gym.Env):
def __init__(
self,
path_to_vm: str,
snapshot_name: str ="init_state",
snapshot_name: str = "init_state",
action_space: str = "computer_13",
tmp_dir: str = "tmp",
cache_dir: str = "cache",
@@ -75,7 +75,7 @@ class DesktopEnv(gym.Env):
self.snapshot_name = snapshot_name
self.tmp_dir_base: str = tmp_dir
self.cache_dir_base: str = cache_dir
self.vm_screen_size = screen_size
self.vm_screen_size = screen_size # todo: add the logic to get the screen size from the VM
self.headless = headless
os.makedirs(self.tmp_dir_base, exist_ok=True)