diff --git a/tests/test_dependency_compatibility.py b/tests/test_dependency_compatibility.py index 4c5e6bfa7..082f8d6bc 100644 --- a/tests/test_dependency_compatibility.py +++ b/tests/test_dependency_compatibility.py @@ -1,7 +1,8 @@ import importlib.util -import pytest import sys +import pytest + def test_httpx_litellm_compatibility(): """Test that litellm is compatible with the latest httpx""" @@ -36,9 +37,9 @@ def test_google_genai_compatibility(): if importlib.util.find_spec("google.generativeai") is None: pytest.skip("google-genai not installed") - from google import generativeai import httpx import litellm + from google import generativeai assert hasattr(generativeai, "version") assert hasattr(httpx, "__version__")