From 7f728c8ebf775097bde949672698c91ac75e71d4 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Mon, 14 Apr 2025 09:40:29 +0000
Subject: [PATCH] fix: Resolve NameError and lint errors in crew_test.py

Co-Authored-By: Joe Moura <joao@crewai.com>
---
 tests/crew_test.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/tests/crew_test.py b/tests/crew_test.py
index 7cb771c3c..9a760b19c 100644
--- a/tests/crew_test.py
+++ b/tests/crew_test.py
@@ -4,11 +4,9 @@ import hashlib
 import json
 import os
 import sys
-
 from concurrent.futures import Future
 from unittest import mock
-from unittest.mock import MagicMock, patch
-
+from unittest.mock import MagicMock, Mock, patch
 
 import pydantic_core
 import pytest
@@ -41,7 +39,6 @@ from crewai.utilities.events.event_listener import EventListener
 from crewai.utilities.rpm_controller import RPMController
 from crewai.utilities.task_output_storage_handler import TaskOutputStorageHandler
 
-
 # Skip streaming tests when running in CI/CD environments
 skip_streaming_in_ci = pytest.mark.skipif(
     os.getenv("CI") is not None, reason="Skipping streaming tests in CI/CD environments"