diff --git a/opencompass/datasets/mbpp.py b/opencompass/datasets/mbpp.py index 99a50ddd..12a6865e 100644 --- a/opencompass/datasets/mbpp.py +++ b/opencompass/datasets/mbpp.py @@ -330,7 +330,11 @@ class MBPPEvaluator(BaseEvaluator): r"'(.*)'\s*\[DONE\]", ] for p in patterns: - match = re.search(p, text, re.DOTALL, timeout=10.0) + try: + match = re.search(p, text, re.DOTALL, timeout=10.0) + except TimeoutError: + match = None + if match: text = match.group(1) break