ver Jan12thv3

debugged
2024-01-13 00:06:11 +08:00
parent e08df57129
commit d4192d3d9c
3 changed files with 6 additions and 5 deletions
--- a/desktop_env/evaluators/metrics/general.py
+++ b/desktop_env/evaluators/metrics/general.py
@@ -150,6 +150,6 @@ def check_accessibility_tree(result: str, rules: Dict[str, Any]) -> float:
 # return 1. - (result is None)

 def run_sqlite3(result: str, rules: Dict[str, Any]) -> float:
-    connection = sqlite3.connect(result)
-    connection.execute(rules["sql"])
-    return float(connection.fetchone()[0] or 0)
+    connection: sqlite3.Connection = sqlite3.connect(result)
+    cursor: sqlite3.Cursor = connection.execute(rules["sql"])
+    return float(cursor.fetchone()[0] or 0)