Skip to content

GitLab

Unverified Commit 04cec4f9 authored by Andreas Karatzas's avatar Andreas Karatzas Committed by GitHub
Browse files

[ROCm][CI] Increase OpenAPI schema test timeouts (#38088) 


Signed-off-by: default avatarAndreas Karatzas <akaratza@amd.com>
parent 14771f71
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 2 deletions
tests/entrypoints/openai/test_openai_schema.py
View file @ 04cec4f9
...@@ -14,14 +14,17 @@ from schemathesis.internal.checks import CheckContext ...@@ -14,14 +14,17 @@ from schemathesis.internal.checks import CheckContext
from schemathesis.models import Case from schemathesis.models import Case
from schemathesis.transports.responses import GenericResponse from schemathesis.transports.responses import GenericResponse
from vllm.platforms import current_platform
from ...utils import RemoteOpenAIServer from ...utils import RemoteOpenAIServer
schemathesis.experimental.OPEN_API_3_1.enable() schemathesis.experimental.OPEN_API_3_1.enable()
MODEL_NAME = "HuggingFaceTB/SmolVLM-256M-Instruct" MODEL_NAME = "HuggingFaceTB/SmolVLM-256M-Instruct"
MAXIMUM_IMAGES = 2 MAXIMUM_IMAGES = 2
DEFAULT_TIMEOUT_SECONDS: Final[int] = 10 _ROCM_TIMEOUT_MULTIPLIER = 3 if current_platform.is_rocm() else 1
LONG_TIMEOUT_SECONDS: Final[int] = 60 DEFAULT_TIMEOUT_SECONDS: Final[int] = 10 * _ROCM_TIMEOUT_MULTIPLIER
LONG_TIMEOUT_SECONDS: Final[int] = 60 * _ROCM_TIMEOUT_MULTIPLIER
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
......
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