diff --git a/tests/crew_test.py b/tests/crew_test.py index 0b8589550..ae5b099c9 100644 --- a/tests/crew_test.py +++ b/tests/crew_test.py @@ -3343,10 +3343,6 @@ def test_crew_testing_function(copy_mock, crew_evaluator_mock): mock.call().print_crew_evaluation_result(), ] ) - mock.call().set_iteration(2), - mock.call().print_crew_evaluation_result(), - ] - ) @pytest.mark.vcr(filter_headers=["authorization"]) diff --git a/tests/utilities/evaluators/test_custom_llm_support.py b/tests/utilities/evaluators/test_custom_llm_support.py index b4281aa66..43f281272 100644 --- a/tests/utilities/evaluators/test_custom_llm_support.py +++ b/tests/utilities/evaluators/test_custom_llm_support.py @@ -1,6 +1,7 @@ -import pytest from unittest.mock import MagicMock +import pytest + from crewai.agent import Agent from crewai.crew import Crew from crewai.llm import LLM