fix counts in testmodelhelpers

exo-explore · Dec 13, 2024 · b0e079b · b0e079b
1 parent 342b5d8
commit b0e079b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/test_model_helpers.py b/test/test_model_helpers.py
@@ -34,7 +34,7 @@ def map_engine(engine):
     expected_models_contains=["llama-3.2-1b", "llama-3.2-3b"],
     min_count=5,
     exact_count=None,
-    max_count=10
+    max_count=15
   ),
   TestCase(
     name="multiple_engines_or",
@@ -58,7 +58,7 @@ def map_engine(engine):
     expected_models_contains=["llama-3.2-1b"],
     min_count=5,
     exact_count=None,
-    max_count=10
+    max_count=15
   ),
   TestCase(
     name="no_engines",
@@ -111,7 +111,7 @@ def _verify_results(self, case, result):
 
     # Special case for distinct lists test
     if case.name == "distinct_engine_lists":
-      self.assertLess(len(result), 10)
+      self.assertLess(len(result), 15)
       self.assertNotIn("mistral-nemo", result)
 
     if case.max_count: