From b0e079b36a477dce6eca7c46a147e2abfad3ce72 Mon Sep 17 00:00:00 2001 From: Alex Cheema Date: Fri, 13 Dec 2024 17:46:44 +0000 Subject: [PATCH] fix counts in testmodelhelpers --- test/test_model_helpers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/test_model_helpers.py b/test/test_model_helpers.py index 8c01104e3..a82150173 100644 --- 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: