Skip to content

GitLab

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
Show 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