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

@ -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(),