tests fix

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

View file

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

View file

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