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

fix

Browse source
This commit is contained in:
Xi Yan 2025-03-13 16:34:29 -07:00
parent fb07642549
commit 7801d9cdf2
2 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/apis/datasets/datasets.py
View file

@ -13,7 +13,8 @@ from llama_stack.apis.resource import Resource, ResourceType
from llama_stack.schema_utils import json_schema_type, register_schema, webmethod
class DatasetPurpose(Enum):
@json_schema_type
class DatasetPurpose(str, Enum):
"""
Purpose of the dataset. Each purpose has a required input data schema.

8
llama_stack/templates/open-benchmark/run.yaml
View file

@ -158,28 +158,28 @@ shields:
- shield_id: meta-llama/Llama-Guard-3-8B
vector_dbs: []
datasets:
- purpose: &id001 eval/messages-answer
- purpose: eval/messages-answer
source:
type: uri
uri: huggingface://llamastack/simpleqa?split=train
metadata: {}
dataset_id: simpleqa
provider_id: huggingface
- purpose: *id001
- purpose: eval/messages-answer
source:
type: uri
uri: huggingface://llamastack/mmlu_cot?split=test&name=all
metadata: {}
dataset_id: mmlu_cot
provider_id: huggingface
- purpose: *id001
- purpose: eval/messages-answer
source:
type: uri
uri: huggingface://llamastack/gpqa_0shot_cot?split=test&name=gpqa_main
metadata: {}
dataset_id: gpqa_cot
provider_id: huggingface
- purpose: *id001
- purpose: eval/messages-answer
source:
type: uri
uri: huggingface://llamastack/math_500?split=test