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

Disable wandb temporarily, speedup the environment step speed by remove useless a11y tree re-get and terminal output

Browse Source
This commit is contained in:
Timothyxxx
2024-03-19 08:57:05 +08:00
parent f992d1f694
commit 8e760fd450
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
desktop_env/envs/desktop_env.py
View File

@@ -285,7 +285,7 @@ class DesktopEnv(gym.Env):
observation = { observation = {
"screenshot": self._get_obs(), "screenshot": self._get_obs(),
"accessibility_tree": self.controller.get_accessibility_tree(), "accessibility_tree": self.controller.get_accessibility_tree(),
"terminal": self.controller.get_terminal_output(), # "terminal": self.controller.get_terminal_output(),
"instruction": self.instruction "instruction": self.instruction
} }

2
lib_run_single.py
View File

@@ -42,7 +42,7 @@ def run_single_example(agent, env, example, max_steps, instruction, args, exampl
screenshot = __f.read() screenshot = __f.read()
_f.write(screenshot) _f.write(screenshot)
# get a11tree and save to wandb # get a11tree and save to wandb
thisrun_a11tree = env.controller.get_accessibility_tree() # thisrun_a11tree = env.controller.get_accessibility_tree()
# str_table.add_data(wandb.Image(data_or_path=os.path.join(example_result_dir, f"step_{step_idx + 1}_{action_timestamp}.png"), caption=f"step_{step_idx + 1}_{action_timestamp}"), # str_table.add_data(wandb.Image(data_or_path=os.path.join(example_result_dir, f"step_{step_idx + 1}_{action_timestamp}.png"), caption=f"step_{step_idx + 1}_{action_timestamp}"),
# thisrun_a11tree, # thisrun_a11tree,
# response, action, action_timestamp, done) # response, action, action_timestamp, done)