From e955ccbd4f49625fce2df90866afc9b62bad698e Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 12 Mar 2025 18:14:33 -0700 Subject: [PATCH] comments --- tests/integration/datasetio/test_datasetio.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/integration/datasetio/test_datasetio.py b/tests/integration/datasetio/test_datasetio.py index 64e187fb5..1885c9bf8 100644 --- a/tests/integration/datasetio/test_datasetio.py +++ b/tests/integration/datasetio/test_datasetio.py @@ -17,13 +17,13 @@ import pytest @pytest.fixture -def test_dataset(llama_stack_client): - register_dataset(llama_stack_client) - yield # This is where the test function will run - +def dataset_for_test(llama_stack_client): + dataset_id = "test_dataset" + register_dataset(llama_stack_client, dataset_id=dataset_id) + yield # Teardown - this always runs, even if the test fails try: - llama_stack_client.datasets.unregister("test_dataset") + llama_stack_client.datasets.unregister(dataset_id) except Exception as e: print(f"Warning: Failed to unregister test_dataset: {e}") @@ -94,7 +94,7 @@ def test_register_unregister_dataset(llama_stack_client): assert len(response) == 0 -def test_get_rows_paginated(llama_stack_client, test_dataset): +def test_get_rows_paginated(llama_stack_client, dataset_for_test): response = llama_stack_client.datasetio.get_rows_paginated( dataset_id="test_dataset", rows_in_page=3,