datasetio

This commit is contained in:
Xi Yan 2025-03-12 16:47:40 -07:00
parent 97d6b87e05
commit 9dcf617556
2 changed files with 7 additions and 3 deletions

View file

@ -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")