danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Prevent HuggingFace network access during tests

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-03-05 12:10:07 +02:00
parent c7b03eb9ab
commit 5e99b84308
No known key found for this signature in database
2 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
.github/workflows/test.yml vendored
View file

@ -68,8 +68,6 @@ jobs:
run: uv run python -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('Qwen/Qwen3-Embedding-0.6B')" run: uv run python -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('Qwen/Qwen3-Embedding-0.6B')"
- name: Run tests with coverage - name: Run tests with coverage
run: uv run pytest -m "not integration" --cov=haiku --cov-report=xml run: uv run pytest -m "not integration" --cov=haiku --cov-report=xml
env:
HF_HUB_OFFLINE: "1"
- name: Upload coverage to Codecov - name: Upload coverage to Codecov
uses: codecov/codecov-action@v5 uses: codecov/codecov-action@v5
with: with:

7
tests/conftest.py
View file

@ -17,13 +17,14 @@ embeddings:
vector_dim: 2560 vector_dim: 2560
""") """)
os.environ["HAIKU_RAG_CONFIG_PATH"] = str(_test_config_path) os.environ["HAIKU_RAG_CONFIG_PATH"] = str(_test_config_path)
os.environ["HF_HUB_OFFLINE"] = "1"
import pydantic_ai.models # noqa: E402 import pydantic_ai.models # noqa: E402
import pytest # noqa: E402 import pytest # noqa: E402
import yaml # noqa: E402 import yaml # noqa: E402
from datasets import Dataset, load_dataset, load_from_disk # noqa: E402
if TYPE_CHECKING: if TYPE_CHECKING:
from datasets import Dataset
from vcr import VCR from vcr import VCR
setattr(pydantic_ai.models, "ALLOW_MODEL_REQUESTS", False) setattr(pydantic_ai.models, "ALLOW_MODEL_REQUESTS", False)
@ -31,7 +32,9 @@ logging.getLogger("vcr.cassette").setLevel(logging.WARNING)
@pytest.fixture(scope="session") @pytest.fixture(scope="session")
def qa_corpus() -> Dataset: def qa_corpus() -> "Dataset":
from datasets import Dataset, load_dataset, load_from_disk
ds_path = Path(__file__).parent / "data" / "dataset" ds_path = Path(__file__).parent / "data" / "dataset"
ds_path.mkdir(parents=True, exist_ok=True) ds_path.mkdir(parents=True, exist_ok=True)
try: try: