huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-19 13:52:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

wip: able to launch gaudi tests

Browse Source
This commit is contained in:
baptiste 2025-04-10 07:52:20 +00:00
parent 919231d2de
commit 6f4e33a1b1
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
integration-tests/conftest.py
View File

@ -94,6 +94,14 @@ def pytest_collection_modifyitems(config, items):
item.add_marker(pytest.mark.skip(reason="requires --gaudi to run"))
selectors.append(skip_not_gaudi)
else:
def skip_gaudi(item):
if "gaudi" in item.keywords:
item.add_marker(pytest.mark.skip(reason="requires --gaudi to run"))
selectors.append(skip_gaudi)
if config.getoption("--neuron"):
def skip_not_neuron(item):
@ -109,12 +117,8 @@ def pytest_collection_modifyitems(config, items):
if "neuron" in item.keywords:
item.add_marker(pytest.mark.skip(reason="requires --neuron to run"))
def skip_gaudi(item):
if "gaudi" in item.keywords:
item.add_marker(pytest.mark.skip(reason="requires --gaudi to run"))
selectors.append(skip_neuron)
selectors.append(skip_gaudi)
for item in items:
for selector in selectors:
selector(item)