From 6130f50b700bc550c773a263a6e362b9d7ad5bba Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Sat, 15 Mar 2025 16:45:28 -0700 Subject: [PATCH] better test --- tests/integration/datasets/test_datasets.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/integration/datasets/test_datasets.py b/tests/integration/datasets/test_datasets.py index 60db95f30..5e4e94a0a 100644 --- a/tests/integration/datasets/test_datasets.py +++ b/tests/integration/datasets/test_datasets.py @@ -70,7 +70,9 @@ def data_url_from_file(file_path: str) -> str: "eval/messages-answer", { "type": "uri", - "uri": data_url_from_file(os.path.join(os.path.dirname(__file__), "test_dataset.csv")), + "uri": data_url_from_file( + os.path.join(os.path.dirname(__file__), "test_dataset.csv") + ), }, "localfs", 5, @@ -84,7 +86,9 @@ def test_register_and_iterrows(llama_stack_client, purpose, source, provider_id, ) assert dataset.identifier is not None assert dataset.provider_id == provider_id - iterrow_response = llama_stack_client.datasets.iterrows(dataset.identifier, limit=limit) + iterrow_response = llama_stack_client.datasets.iterrows( + dataset.identifier, limit=limit + ) assert len(iterrow_response.data) == limit dataset_list = llama_stack_client.datasets.list()