phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-10 04:08:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

pre

Browse source
This commit is contained in:
Xi Yan 2025-03-15 16:46:37 -07:00
parent 6130f50b70
commit 956503535f
1 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/integration/datasets/test_datasets.py
View file

@ -70,9 +70,7 @@ def data_url_from_file(file_path: str) -> str:
"eval/messages-answer", "eval/messages-answer",
{ {
"type": "uri", "type": "uri",
"uri": data_url_from_file( "uri": data_url_from_file(os.path.join(os.path.dirname(__file__), "test_dataset.csv")),
os.path.join(os.path.dirname(__file__), "test_dataset.csv")
),
}, },
"localfs", "localfs",
5, 5,
@ -86,9 +84,7 @@ def test_register_and_iterrows(llama_stack_client, purpose, source, provider_id,
) )
assert dataset.identifier is not None assert dataset.identifier is not None
assert dataset.provider_id == provider_id assert dataset.provider_id == provider_id
iterrow_response = llama_stack_client.datasets.iterrows( iterrow_response = llama_stack_client.datasets.iterrows(dataset.identifier, limit=limit)
dataset.identifier, limit=limit
)
assert len(iterrow_response.data) == limit assert len(iterrow_response.data) == limit
dataset_list = llama_stack_client.datasets.list() dataset_list = llama_stack_client.datasets.list()