From 3f6c1dc57c7c29c0790fb1b8ecdb72a7bb5a1bec Mon Sep 17 00:00:00 2001 From: Nigel Jones Date: Wed, 11 Mar 2026 15:03:24 +0000 Subject: [PATCH] fix: add missing requires_gpu and requires_heavy_ram markers to HuggingFace metrics test test_huggingface_token_metrics_integration was missing the markers that trigger conftest auto-skip logic on systems without sufficient GPU or RAM (threshold: 48GB). Without them the test runs unconditionally, loading a full HF model and consuming excessive time and memory. Fixes #620 --- test/telemetry/test_metrics_backend.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/test/telemetry/test_metrics_backend.py b/test/telemetry/test_metrics_backend.py index 9bba4503a..069d3f26f 100644 --- a/test/telemetry/test_metrics_backend.py +++ b/test/telemetry/test_metrics_backend.py @@ -297,6 +297,12 @@ async def test_litellm_token_metrics_integration( @pytest.mark.asyncio @pytest.mark.llm @pytest.mark.huggingface +@pytest.mark.requires_gpu +@pytest.mark.requires_heavy_ram +@pytest.mark.skipif( + int(os.environ.get("CICD", 0)) == 1, + reason="Skipping HuggingFace metrics test in CI - requires GPU and model download", +) @pytest.mark.parametrize("stream", [False, True], ids=["non-streaming", "streaming"]) async def test_huggingface_token_metrics_integration( enable_metrics, metric_reader, stream, gh_run