Fix benchmark for QuestionValidator

road-core · Feb 4, 2025 · ef89ec1 · ef89ec1
1 parent 7edfdf3
commit ef89ec1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/benchmarks/test_question_validator.py b/tests/benchmarks/test_question_validator.py
@@ -24,7 +24,7 @@ def perform_question_validation_benchmark(benchmark, question):
     # be performed
     llm_loader = mock_llm_loader(
         None,
-        expected_params=("p1", "m1", {GenericLLMParameters.MAX_TOKENS_FOR_RESPONSE: 4}),
+        expected_params=("p1", "m1", {GenericLLMParameters.MAX_TOKENS_FOR_RESPONSE: 4}, False),
     )
 
     # check that LLM loader was called with expected parameters

diff --git a/tests/mock_classes/mock_llm_loader.py b/tests/mock_classes/mock_llm_loader.py
@@ -27,7 +27,7 @@ def loader(*args, **kwargs):
         # if expected params are provided, check if (mocked) LLM loader
         # was called with expected parameters
         if expected_params is not None:
-            assert expected_params == args, expected_params
+            assert expected_params == args, (expected_params, args)
         return MockLLMLoader(llm)
 
     return loader