Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "0a161121852ee5f34b939279d54b5d3e231ca501"
Unverified Commit 8751fb35 authored by Kiersten Stokes's avatar Kiersten Stokes Committed by GitHub
Browse files

Update unitxt task.py to bring in line with recent repo changes (#2684)

parent 684fd2dd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
lm_eval/tasks/unitxt/task.py
View file @ 8751fb35
...@@ -109,6 +109,7 @@ class Unitxt(ConfigurableTask): ...@@ -109,6 +109,7 @@ class Unitxt(ConfigurableTask):
apply_chat_template: bool = False, apply_chat_template: bool = False,
fewshot_as_multiturn: bool = False, fewshot_as_multiturn: bool = False,
chat_template: Optional[Callable] = None, chat_template: Optional[Callable] = None,
gen_prefix: Optional[str] = None,
) -> str: ) -> str:
source = self.doc_to_text(doc) source = self.doc_to_text(doc)
if isinstance(source, list): if isinstance(source, list):
...@@ -134,6 +135,7 @@ class Unitxt(ConfigurableTask): ...@@ -134,6 +135,7 @@ class Unitxt(ConfigurableTask):
part of the document for `doc`. part of the document for `doc`.
""" """
kwargs.pop("apply_chat_template", False) # Not used by unitxt kwargs.pop("apply_chat_template", False) # Not used by unitxt
kwargs.pop("chat_template", False) # Not used by unitxt
return [ return [
Instance( Instance(
request_type="generate_until", request_type="generate_until",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment