Skip to content

Commit 2a47dc8

Browse files
Minor change in the test file
1 parent c531a18 commit 2a47dc8

1 file changed

Lines changed: 3 additions & 8 deletions

File tree

tests/benchmarks/micro_benchmarks/test_huggingface_e2e.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,25 +4,20 @@
44
"""End-to-end integration tests for HuggingFace model loading.
55
66
These tests actually download and load models from HuggingFace Hub.
7-
They are marked with @pytest.mark.integration and skipped by default
8-
in CI due to network/disk requirements.
9-
10-
Run with: pytest -m integration tests/benchmarks/micro_benchmarks/test_huggingface_e2e.py
7+
Skipped when SB_TEST_CUDA=0.
118
"""
129

1310
import pytest
1411
import torch
1512

1613
transformers = pytest.importorskip('transformers')
1714

15+
from tests.helper import decorator
1816
from superbench.benchmarks.micro_benchmarks.huggingface_model_loader import HuggingFaceModelLoader
1917
from superbench.benchmarks.micro_benchmarks.model_source_config import ModelSourceConfig
2018

2119

22-
# Skip all tests if no GPU available (most HF models need significant resources)
23-
pytestmark = pytest.mark.integration
24-
25-
20+
@decorator.cuda_test
2621
class TestHuggingFaceE2E:
2722
"""End-to-end tests for HuggingFace model loading."""
2823

0 commit comments

Comments
 (0)