diff --git a/tests/test_timeout/test_agent_timeout.py b/tests/test_timeout/test_agent_timeout.py index 71ab86d9c..967a75925 100644 --- a/tests/test_timeout/test_agent_timeout.py +++ b/tests/test_timeout/test_agent_timeout.py @@ -1,9 +1,11 @@ -import pytest import time from unittest.mock import MagicMock, patch +import pytest + from crewai import Agent, Task + def test_agent_max_execution_time(): """Test that max_execution_time parameter is enforced.""" # Create a simple test function that will be used to simulate a long-running task