From 8dc54d79ff0422a7808d63e4b80b7928839d6dc8 Mon Sep 17 00:00:00 2001 From: Greyson LaLonde Date: Tue, 20 Jan 2026 12:59:57 -0500 Subject: [PATCH] chore: fix existing flaky test --- lib/crewai/tests/utilities/test_events.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/lib/crewai/tests/utilities/test_events.py b/lib/crewai/tests/utilities/test_events.py index f637b6c3d..8d70fe57f 100644 --- a/lib/crewai/tests/utilities/test_events.py +++ b/lib/crewai/tests/utilities/test_events.py @@ -348,9 +348,7 @@ def test_agent_emits_execution_error_event(base_agent, base_task): error_message = "Error happening while sending prompt to model." base_agent.max_retry_limit = 0 - with patch.object( - CrewAgentExecutor, "invoke", wraps=base_agent.agent_executor.invoke - ) as invoke_mock: + with patch.object(CrewAgentExecutor, "invoke") as invoke_mock: invoke_mock.side_effect = Exception(error_message) with pytest.raises(Exception): # noqa: B017