From e7c5dc289a498c0c086987cce86b33ecfc12c5a7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A9mi=20Louf?= <remi@thetypicalset.com>
Date: Wed, 16 Oct 2024 15:29:20 +0200
Subject: [PATCH] Increase timeout for lm-format-enforcer

---
 src/benchmark_lfe.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/benchmark_lfe.py b/src/benchmark_lfe.py
index 7379de8..22a62cf 100644
--- a/src/benchmark_lfe.py
+++ b/src/benchmark_lfe.py
@@ -11,7 +11,7 @@
 class LMFormatEnforcerRegex:
     params = [models, regex_cases.keys()]
     param_names = ["model", "regex_name"]
-    timeout = 600
+    timeout = 1200
 
     def setup(self, model, _):
         """Set up the benchmark.
@@ -44,7 +44,7 @@ def teardown(self, *args):
 class LMFormatEnforcerJsonSchema:
     params = [models, json_cases.keys()]
     param_names = ["model", "json_schema_name"]
-    timeout = 600
+    timeout = 1200
 
     def setup(self, model, _):
         """Set up the benchmark.