diff --git a/tests/crew_test.py b/tests/crew_test.py index f1706b210..8dc58c352 100644 --- a/tests/crew_test.py +++ b/tests/crew_test.py @@ -3102,7 +3102,8 @@ def test_crew_testing_function(kickoff_mock, copy_mock, crew_evaluator): copy_mock.return_value = crew n_iterations = 2 - crew.test(n_iterations, eval_llm="gpt-4o-mini", inputs={"topic": "AI"}) + llm_instance = LLM('gpt-4o-mini') + crew.test(n_iterations, llm_instance, inputs={"topic": "AI"}) # Ensure kickoff is called on the copied crew kickoff_mock.assert_has_calls( @@ -3111,7 +3112,7 @@ def test_crew_testing_function(kickoff_mock, copy_mock, crew_evaluator): crew_evaluator.assert_has_calls( [ - mock.call(crew,LLM('gpt-4o-mini')), + mock.call(crew,llm_instance), mock.call().set_iteration(1), mock.call().set_iteration(2), mock.call().print_crew_evaluation_result(),