This commit is contained in:
Xi Yan 2025-03-12 18:15:15 -07:00
parent e955ccbd4f
commit 59b1263090

View file

@ -20,7 +20,7 @@ import pytest
def dataset_for_test(llama_stack_client): def dataset_for_test(llama_stack_client):
dataset_id = "test_dataset" dataset_id = "test_dataset"
register_dataset(llama_stack_client, dataset_id=dataset_id) register_dataset(llama_stack_client, dataset_id=dataset_id)
yield yield
# Teardown - this always runs, even if the test fails # Teardown - this always runs, even if the test fails
try: try:
llama_stack_client.datasets.unregister(dataset_id) llama_stack_client.datasets.unregister(dataset_id)