Skip to content

Commit

Permalink
Remove client tests that depend on HF API
Browse files Browse the repository at this point in the history
  • Loading branch information
tgaddair committed Jan 8, 2024
1 parent c2762af commit 09c1c6e
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 201 deletions.
51 changes: 0 additions & 51 deletions clients/python/tests/conftest.py
Original file line number Diff line number Diff line change
@@ -1,51 +0,0 @@
import pytest

from lorax import __version__
from huggingface_hub.utils import build_hf_headers


@pytest.fixture
def flan_t5_xxl():
return "google/flan-t5-xxl"


@pytest.fixture
def fake_model():
return "fake/model"


@pytest.fixture
def unsupported_model():
return "gpt2"


@pytest.fixture
def base_url():
return "https://api-inference.huggingface.co/models"


@pytest.fixture
def bloom_url(base_url, bloom_model):
return f"{base_url}/{bloom_model}"


@pytest.fixture
def flan_t5_xxl_url(base_url, flan_t5_xxl):
return f"{base_url}/{flan_t5_xxl}"


@pytest.fixture
def fake_url(base_url, fake_model):
return f"{base_url}/{fake_model}"


@pytest.fixture
def unsupported_url(base_url, unsupported_model):
return f"{base_url}/{unsupported_model}"


@pytest.fixture(scope="session")
def hf_headers():
return build_hf_headers(
library_name="lorax-tests", library_version=__version__
)
150 changes: 0 additions & 150 deletions clients/python/tests/test_client.py

This file was deleted.

0 comments on commit 09c1c6e

Please sign in to comment.