-
Notifications
You must be signed in to change notification settings - Fork 149
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Remove client tests that depend on HF API
- Loading branch information
Showing
2 changed files
with
0 additions
and
201 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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__ | ||
) | ||
This file was deleted.
Oops, something went wrong.