phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-16 18:12:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

tests fix

Browse source
This commit is contained in:
Xi Yan 2024-11-13 11:13:32 -05:00
parent 4b77d26ece
commit f31476541e
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/tests/datasetio/test_datasetio.py
View file

@ -57,7 +57,7 @@ async def register_dataset(
await datasets_impl.register_dataset( await datasets_impl.register_dataset(
dataset_id=dataset_id, dataset_id=dataset_id,
schema=dataset_schema, dataset_schema=dataset_schema,
url=URL(uri=test_url), url=URL(uri=test_url),
) )

2
llama_stack/providers/tests/eval/test_eval.py
View file

@ -163,7 +163,7 @@ class Testeval:
await datasets_impl.register_dataset( await datasets_impl.register_dataset(
dataset_id="mmlu", dataset_id="mmlu",
schema={ dataset_schema={
"input_query": StringType(), "input_query": StringType(),
"expected_answer": StringType(), "expected_answer": StringType(),
"chat_completion_input": ChatCompletionInputType(), "chat_completion_input": ChatCompletionInputType(),