From 2fc30a92ec76ef2cd4e6e736b640b3d9c015b0d9 Mon Sep 17 00:00:00 2001 From: Francesco Bertolotti Date: Wed, 28 May 2025 10:42:32 +0200 Subject: [PATCH] TimeOutError exception handling --- opencompass/datasets/mbpp.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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