From f8b2ba0860d2a57a1a3f1d1034db479ea248bde6 Mon Sep 17 00:00:00 2001 From: Andrei Alexandru Date: Wed, 20 Dec 2023 12:01:55 +0000 Subject: [PATCH] Fix small typo in oaieval run function --- evals/cli/oaieval.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/evals/cli/oaieval.py b/evals/cli/oaieval.py index 20b7d4c3bf..65db6fc184 100644 --- a/evals/cli/oaieval.py +++ b/evals/cli/oaieval.py @@ -7,8 +7,6 @@ import sys from typing import Any, Mapping, Optional, Union, cast -import openai - import evals import evals.api import evals.base @@ -137,11 +135,11 @@ def run(args: OaiEvalArguments, registry: Optional[Registry] = None) -> str: # If the user provided an argument to --completion_args, parse it into a dict here, to be passed to the completion_fn creation **kwargs completion_args = args.completion_args.split(",") - additonal_completion_args = {k: v for k, v in (kv.split("=") for kv in completion_args if kv)} + additional_completion_args = {k: v for k, v in (kv.split("=") for kv in completion_args if kv)} completion_fns = args.completion_fn.split(",") completion_fn_instances = [ - registry.make_completion_fn(url, **additonal_completion_args) for url in completion_fns + registry.make_completion_fn(url, **additional_completion_args) for url in completion_fns ] run_config = {