Merge branch 'eval-for-new-labelizer' of github.com:lavague-ai/LaVagu…

…e into eval-for-new-labelizer
lavague-ai · Aug 29, 2024 · 3adbf95 · 3adbf95
2 parents 80cb83d + 4ffd760
commit 3adbf95
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"])