mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-31 03:50:00 +00:00
24 lines
637 B
Python
24 lines
637 B
Python
# 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.
|
|
|
|
import base64
|
|
import mimetypes
|
|
import os
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
# How to run this test:
|
|
#
|
|
# LLAMA_STACK_CONFIG="template-name" pytest -v tests/integration/datasets
|
|
|
|
|
|
def test_register_dataset(llama_stack_client):
|
|
dataset = llama_stack_client.datasets.register(
|
|
purpose="eval/messages-answer",
|
|
source={"type": "uri", "uri": "huggingface://llamastack/simpleqa?split=train"},
|
|
)
|
|
print(dataset)
|