modified: code/judge.py

GAIR-NLP · Jun 17, 2024 · 249aa49 · 249aa49
1 parent 2949429
commit 249aa49
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/code/judge.py b/code/judge.py
@@ -67,16 +67,14 @@ def extract_boxed_answer(self, text):
                 elif text[end_index] == '}':
                     stack -= 1
                 end_index += 1
-                if stack == 0:
-                    content = text[start_index:end_index - 1]
-                    if not content:
-                        return text
-                    else:
-                        content = self.normalize_answer(content)
-                        return content
-            return text
-        else:
-            return text
+            if stack == 0:
+                content = text[start_index:end_index - 1]
+                if not content:
+                    return text
+                else:
+                    content = self.normalize_answer(content)
+                    return content
+        return text
 
 
     def split_by_comma(self, expr: str):