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):
dataset_id = "test_dataset"
register_dataset(llama_stack_client, dataset_id=dataset_id)
yield
yield
# Teardown - this always runs, even if the test fails
try:
llama_stack_client.datasets.unregister(dataset_id)