diff --git a/tests/crew_test.py b/tests/crew_test.py index ceacfba8c..9296d4503 100644 --- a/tests/crew_test.py +++ b/tests/crew_test.py @@ -3,9 +3,13 @@ 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 Mock, MagicMock + import pydantic_core import pytest @@ -39,9 +43,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 -import sys -from unittest.mock import Mock, MagicMock - # Skip streaming tests when running in CI/CD environments skip_streaming_in_ci = pytest.mark.skipif(