From 329164ba09d7e6b88e25fe0f2a1211aa81809853 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 12 Mar 2025 22:28:19 -0700 Subject: [PATCH] add dataset_id --- tests/integration/scoring/test_scoring.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/integration/scoring/test_scoring.py b/tests/integration/scoring/test_scoring.py index 3f0f832c1..970a96f40 100644 --- a/tests/integration/scoring/test_scoring.py +++ b/tests/integration/scoring/test_scoring.py @@ -12,12 +12,13 @@ from ..datasetio.test_datasetio import register_dataset @pytest.fixture def rag_dataset_for_test(llama_stack_client): - register_dataset(llama_stack_client, for_rag=True) + dataset_id = "test_dataset" + register_dataset(llama_stack_client, for_rag=True, dataset_id=dataset_id) 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") + llama_stack_client.datasets.unregister(dataset_id) except Exception as e: print(f"Warning: Failed to unregister test_dataset: {e}")