alihan/exo
1
0
Fork 0
mirror of https://github.com/exo-explore/exo.git synced 2025-10-23 02:57:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix counts in testmodelhelpers

Browse Source
This commit is contained in:
Alex Cheema
2024-12-13 17:46:44 +00:00
parent 342b5d8ac0
commit b0e079b36a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_model_helpers.py
View File

@@ -34,7 +34,7 @@ test_cases = [
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 @@ test_cases = [
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 @@ class TestModelHelpers(unittest.TestCase):
# 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: