fdaf4e31bd
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com> Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
|
|
"""Shared, importable test helper utilities.
|
|
|
|
Submodules (``assertions``, ``clean``, ``client``, ``media``, ``runtime``, …) are imported
|
|
explicitly by callers. Unit tests for these helpers live under ``tests/``.
|
|
Avoid star-importing everything here: that ran before refactor only inside the
|
|
old monolithic ``conftest``; a greedy ``__init__`` changes import order and can
|
|
affect in-process Omni (``OmniRunner`` / offline e2e) vs subprocess-based
|
|
``OmniServer`` tests.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
|
|
def skip_if_gated_repo_inaccessible(
|
|
repo_id: str,
|
|
*,
|
|
revision: str | None = None,
|
|
filename: str = "config.json",
|
|
) -> None:
|
|
"""Skip the test if a gated HuggingFace repo is not accessible.
|
|
|
|
Tries to download ``filename`` via ``hf_hub_download``, which performs an
|
|
actual file-access check (unlike ``HfApi().model_info()``
|
|
that only checks metadata). If the token has metadata access but not
|
|
file-download access, ``hf_hub_download`` will raise ``GatedRepoError``
|
|
and we skip cleanly.
|
|
"""
|
|
try:
|
|
from huggingface_hub.errors import GatedRepoError, RepositoryNotFoundError
|
|
|
|
from vllm_omni.transformers_utils.repo_utils import hf_api
|
|
except Exception:
|
|
return
|
|
try:
|
|
hf_api().hf_hub_download(repo_id=repo_id, filename=filename, revision=revision)
|
|
except GatedRepoError as exc:
|
|
import pytest
|
|
|
|
pytest.skip(
|
|
f"Skipping: gated HF repo {repo_id!r} inaccessible to the current "
|
|
f"HF_TOKEN ({exc}). See docs/contributing/ci/hf_credentials.md."
|
|
)
|
|
except RepositoryNotFoundError as exc:
|
|
import pytest
|
|
|
|
pytest.skip(f"Skipping: HF repo {repo_id!r} not found ({exc}).")
|
|
except Exception:
|
|
return
|