From 616ffe2aba093eb2998d889a3895b45d9f1cd55a Mon Sep 17 00:00:00 2001 From: Eduardo Chiarotti Date: Thu, 25 Jul 2024 15:30:54 -0300 Subject: [PATCH] feat: fix test --- tests/crew_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/crew_test.py b/tests/crew_test.py index 61e3ffc52..e82913c69 100644 --- a/tests/crew_test.py +++ b/tests/crew_test.py @@ -2516,7 +2516,7 @@ def test_crew_testing_function(mock_kickoff, crew_evaluator): tasks=[task], ) n_iterations = 2 - crew.test(n_iterations, model="gpt-4o-mini", inputs={"topic": "AI"}) + crew.test(n_iterations, openai_model_name="gpt-4o-mini", inputs={"topic": "AI"}) assert len(mock_kickoff.mock_calls) == n_iterations mock_kickoff.assert_has_calls(