fix patch extraction

ASSERT-KTH · Sep 4, 2024 · 961ef22 · 961ef22
1 parent 566d7c8
commit 961ef22
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/elleelleaime/evaluate/strategies/google/google.py b/elleelleaime/evaluate/strategies/google/google.py
@@ -24,6 +24,7 @@ def _evaluate_impl(self, bug: Bug, sample: dict) -> Optional[List[dict]]:
         for generation in sample["generation"]:
             for candidate in generation["candidates"]:
                 candidate_patch = candidate["content"]["parts"][0]["text"]
+                candidate_patch = self.extract_patch_from_message(candidate_patch)
                 evaluation.append(
                     self.evaluate_generation(bug, sample, candidate_patch)
                 )