diff --git a/tests/integration/datasets/test_script.py b/tests/integration/datasets/test_script.py deleted file mode 100644 index a1587a09e..000000000 --- a/tests/integration/datasets/test_script.py +++ /dev/null @@ -1,43 +0,0 @@ -# Copyright (c) Meta Platforms, Inc. and affiliates. -# All rights reserved. -# -# This source code is licensed under the terms described in the LICENSE file in -# the root directory of this source tree. - -from llama_stack_client import LlamaStackClient -from rich.pretty import pprint - - -def test_register_dataset(): - client = LlamaStackClient(base_url="http://localhost:8321") - # dataset = client.datasets.register( - # purpose="eval/messages-answer", - # source={ - # "type": "uri", - # "uri": "huggingface://datasets/llamastack/simpleqa?split=train", - # }, - # ) - dataset = client.datasets.register( - purpose="eval/messages-answer", - source={ - "type": "rows", - "rows": [ - { - "messages": [{"role": "user", "content": "Hello, world!"}], - "answer": "Hello, world!", - }, - { - "messages": [{"role": "user", "content": "What is the capital of France?"}], - "answer": "Paris", - }, - ], - }, - ) - dataset_id = dataset.identifier - pprint(dataset) - rows = client.datasets.iterrows(dataset_id=dataset_id, limit=10) - pprint(rows) - - -if __name__ == "__main__": - test_register_dataset()