diff --git a/tests/test_kickoff_for_each_parallel.py b/tests/test_kickoff_for_each_parallel.py index cc20cb9d1..de4890ebe 100644 --- a/tests/test_kickoff_for_each_parallel.py +++ b/tests/test_kickoff_for_each_parallel.py @@ -1,8 +1,9 @@ """Test for the kickoff_for_each_parallel method in Crew class.""" import concurrent.futures +from unittest.mock import MagicMock, patch + import pytest -from unittest.mock import patch, MagicMock from crewai.agent import Agent from crewai.crew import Crew