From 2fc30a92ec76ef2cd4e6e736b640b3d9c015b0d9 Mon Sep 17 00:00:00 2001
From: Francesco Bertolotti <francesco.bertolotti@igenius.ai>
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