From 92101e77e424e9f5deb4822836a173184dd1cddb Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 12 Feb 2025 22:03:35 +0000 Subject: [PATCH] style: Fix remaining linting issues Co-Authored-By: Joe Moura --- tests/cli/provider_test.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/cli/provider_test.py b/tests/cli/provider_test.py index 5ff435dcb..cfa4b98ea 100644 --- a/tests/cli/provider_test.py +++ b/tests/cli/provider_test.py @@ -1,8 +1,9 @@ -from unittest.mock import Mock, patch +import json +import time import pytest import requests -import time +from unittest.mock import Mock, patch from crewai.cli.constants import JSON_URL, MODELS, PROVIDERS from crewai.cli.provider import fetch_provider_data, get_provider_data @@ -27,6 +28,7 @@ def test_fetch_provider_data_success(): mock_response = Mock() mock_response.headers = {'content-type': 'application/json'} mock_response.json.return_value = mock_data + mock_response.iter_content.return_value = [json.dumps(mock_data).encode()] mock_get.return_value = mock_response result = fetch_provider_data('/tmp/cache.json') assert result == mock_data