destroy in right function

lavague-ai · Aug 29, 2024 · 4ffd760 · 4ffd760
1 parent 4e507c8
commit 4ffd760
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lavague-core/lavague/core/evaluator.py b/lavague-core/lavague/core/evaluator.py
@@ -138,6 +138,8 @@ def evaluate(
                     print("ERROR: ", i)
                     traceback.print_exc()
                     nodes = []
+                if driver:
+                    driver.destroy()
                 nodes = "\n".join(nodes)
                 results.at[i, "result_nodes"] = nodes
                 results.at[i, "recall"] = (
@@ -208,8 +210,6 @@ def evaluate(
                     print("ERROR: ", i)
                     traceback.print_exc()
                     test_action = FAIL_ACTION
-                if driver:
-                    driver.destroy()
                 results.at[i, "correct_action"] = action["name"] == test_action["name"]
                 results.at[i, "correct_xpath"] = (
                     normalize_xpath(action["args"]["xpath"])