diff --git a/.github/workflows/test-proxy-e2e-azure-batches.yml b/.github/workflows/test-proxy-e2e-azure-batches.yml index 38d436dc1f..4d74f3db0a 100644 --- a/.github/workflows/test-proxy-e2e-azure-batches.yml +++ b/.github/workflows/test-proxy-e2e-azure-batches.yml @@ -56,7 +56,7 @@ jobs: run: | poetry config virtualenvs.in-project true poetry install --with dev,proxy-dev --extras "proxy" - poetry run pip install psycopg2-binary uvicorn fastapi httpx + poetry run pip install psycopg2-binary uvicorn fastapi httpx tenacity - name: Setup litellm-enterprise run: | diff --git a/tests/proxy_e2e_azure_batches_tests/test_managed_files_base.py b/tests/proxy_e2e_azure_batches_tests/test_managed_files_base.py index a55818ce69..79e7e58f39 100644 --- a/tests/proxy_e2e_azure_batches_tests/test_managed_files_base.py +++ b/tests/proxy_e2e_azure_batches_tests/test_managed_files_base.py @@ -12,8 +12,6 @@ import httpx import openai import psycopg2 import pytest - -tenacity = pytest.importorskip("tenacity", reason="tenacity required for e2e batch tests") from tenacity import Retrying, stop_after_delay, wait_fixed sys.path.insert(0, os.path.abspath("../..")) diff --git a/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py b/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py index db37674ac9..262c55efc5 100644 --- a/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py +++ b/tests/proxy_e2e_azure_batches_tests/test_proxy_e2e_azure_batches.py @@ -7,8 +7,6 @@ import warnings import httpx import openai import pytest - -tenacity = pytest.importorskip("tenacity", reason="tenacity required for e2e batch tests") from tenacity import RetryError sys.path.insert(0, os.path.abspath("../.."))