"vllm/vscode:/vscode.git/clone" did not exist on "ff68035932431c1892563f823735ee350336036f"
Unverified Commit 0aa1c1bb authored by Dmitry Tokarev's avatar Dmitry Tokarev Committed by GitHub
Browse files

fix: openai import in FT migration tests (#8594)


Signed-off-by: default avatarDmitry Tokarev <dtokarev@nvidia.com>
Co-authored-by: default avatarcoderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
parent b8e32933
......@@ -21,6 +21,7 @@ mistral-common>=1.10.0
mypy==1.18.2
# For NATS object store verification in router tests
nats-py==2.12.0
openai==2.32.0 # for Fault Tolerance migration tests
psutil<=7.0.0 # System package, may vary by platform (was >=5.0.0)
# Required by pyproject.toml warning filters (pydantic.warnings.PydanticDeprecatedSince20)
pydantic>=2.11.4,<2.13
......
......@@ -8,6 +8,7 @@ import time
import pytest
import requests
from openai import APIError, OpenAI
from tests.utils.constants import FAULT_TOLERANCE_MODEL_NAME
from tests.utils.managed_process import (
......@@ -15,12 +16,6 @@ from tests.utils.managed_process import (
)
from tests.utils.managed_process import ManagedProcess, terminate_process_tree
openai = pytest.importorskip(
"openai", reason="openai package is required for fault tolerance migration tests"
)
APIError = openai.APIError
OpenAI = openai.OpenAI
logger = logging.getLogger(__name__)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment