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 79e7e58f39..a55818ce69 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,6 +12,8 @@ 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 262c55efc5..db37674ac9 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,6 +7,8 @@ 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("../.."))