mirror of
https://github.com/meta-llama/llama-stack.git
synced 2026-01-03 03:42:15 +00:00
datasetio
This commit is contained in:
parent
97d6b87e05
commit
9dcf617556
2 changed files with 7 additions and 3 deletions
|
|
@ -15,11 +15,12 @@ import pytest
|
|||
#
|
||||
# LLAMA_STACK_CONFIG="template-name" pytest -v tests/integration/datasetio
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def test_dataset(llama_stack_client):
|
||||
register_dataset(llama_stack_client)
|
||||
yield # This is where the test function will run
|
||||
|
||||
|
||||
# Teardown - this always runs, even if the test fails
|
||||
try:
|
||||
llama_stack_client.datasets.unregister("test_dataset")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue