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

patch: fix the bug when expected getter is none

Browse Source
This commit is contained in:
MillanK0817
2025-04-08 15:35:29 +08:00
parent bbfeecb475
commit eb24584098
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
desktop_env/desktop_env.py
View File

@@ -288,7 +288,7 @@ class DesktopEnv(gym.Env):
if self.metric_conj == 'and': if self.metric_conj == 'and':
return 0 return 0
if "expected" in self.evaluator: if "expected" in self.evaluator and self.expected_getter and self.evaluator["expected"]:
expected_state = self.expected_getter[idx](self, self.evaluator["expected"][idx]) expected_state = self.expected_getter[idx](self, self.evaluator["expected"][idx])
metric: int = metric(result_state, expected_state, **self.metric_options[idx]) metric: int = metric(result_state, expected_state, **self.metric_options[idx])
else: else:
@@ -310,7 +310,7 @@ class DesktopEnv(gym.Env):
logger.error("File not found!") logger.error("File not found!")
return 0 return 0
if "expected" in self.evaluator: if "expected" in self.evaluator and self.expected_getter and self.evaluator["expected"]:
expected_state = self.expected_getter(self, self.evaluator["expected"]) expected_state = self.expected_getter(self, self.evaluator["expected"])
metric: float = self.metric(result_state, expected_state, **self.metric_options) metric: float = self.metric(result_state, expected_state, **self.metric_options)
else: else: