修复关于abstract的搜索

2023-06-17 19:27:21 +08:00 · 2023-06-17 19:27:21 +08:00 · 9a5a509dd9
commit 9a5a509dd9
parent cbcb98ef6a
1 changed files with 3 additions and 1 deletions
--- a/crazy_functions/latex_utils.py
+++ b/crazy_functions/latex_utils.py
@ -424,7 +424,9 @@ class LatexPaperSplit():
        if mode == 'translate_zh':
            pattern = re.compile(r'\\begin\{abstract\}.*\n')
            match = pattern.search(result_string)
-            assert match is not None, "Cannot find paper abstract section!"
+            if not match:
+                pattern = re.compile(r'\\abstract\{')
+                match = pattern.search(result_string)
            position = match.end()
            result_string = result_string[:position] + self.msg + msg + self.msg_declare + result_string[position:]
        return result_string