From 06d667dde12b61ad3e2000f6beb607f2b562e6a2 Mon Sep 17 00:00:00 2001 From: "Yao, Qing" Date: Thu, 5 Sep 2024 09:52:21 +0800 Subject: [PATCH] Fix for bigcode eval UT Signed-off-by: Yao, Qing --- evals/evaluation/bigcode_evaluation_harness/arguments.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/evals/evaluation/bigcode_evaluation_harness/arguments.py b/evals/evaluation/bigcode_evaluation_harness/arguments.py index b2303a47..8683873a 100644 --- a/evals/evaluation/bigcode_evaluation_harness/arguments.py +++ b/evals/evaluation/bigcode_evaluation_harness/arguments.py @@ -258,6 +258,7 @@ def __init__( check_references=False, user_model=None, # used for pass model object tokenizer=None, # used for pass tokenizer object + codegen_url=None, ): self.prefix = prefix self.do_sample = do_sample @@ -300,3 +301,4 @@ def __init__( self.check_references = check_references self.user_model = user_model self.tokenizer = tokenizer + self.codegen_url = codegen_url