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

Merge branch 'main' of https://github.com/ztjhz/DesktopEnv

Browse Source
This commit is contained in:
tsuky_chen
2024-01-16 01:07:54 +08:00
parent 9fe3a5db3b 28d8c0c528
commit 91823b1bcd
10 changed files with 82 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
desktop_env/envs/desktop_env.py
View File

@@ -231,8 +231,11 @@ class DesktopEnv(gym.Env):
# the set of all possible actions defined in the action representation
self.controller.execute_action(action)
elif self.action_space == "pyautogui":
# the set of all possible python commands insides `pyautogui`
self.controller.execute_python_command(action)
if action in ['WAIT', 'FAIL', 'DONE']:
self.controller.execute_action(action)
else:
# the set of all possible python commands insides `pyautogui`
self.controller.execute_python_command(action)
observation = {
"screenshot": self._get_obs(),