Commit 898e0ace authored by Baber's avatar Baber
Browse files

nit

parent 37f17c3a
...@@ -822,8 +822,12 @@ class ConfigurableTask(Task): ...@@ -822,8 +822,12 @@ class ConfigurableTask(Task):
self.download(self.config.dataset_kwargs) self.download(self.config.dataset_kwargs)
else: else:
self.dataset = self.config.download_dataset( self.dataset = self.config.download_dataset(
self.config.metadata.get( metadata=self.config.metadata.get(
"tokenizer", self.config.metadata.get("pretrained") "tokenizer",
self.config.metadata.get("pretrained"),
**self.config.dataset_kwargs
if self.config.dataset_kwargs is not None
else {},
) )
) )
self._training_docs = None self._training_docs = None
...@@ -931,7 +935,9 @@ class ConfigurableTask(Task): ...@@ -931,7 +935,9 @@ class ConfigurableTask(Task):
f'Both target_delimiter "{self.config.target_delimiter}" and target choice: "{choice}" do not have whitespace, ignore if the language you are evaluating on does not require/use whitespace' f'Both target_delimiter "{self.config.target_delimiter}" and target choice: "{choice}" do not have whitespace, ignore if the language you are evaluating on does not require/use whitespace'
) )
def download(self, dataset_kwargs: Optional[Dict[str, Any]] = None) -> None: def download(
self, dataset_kwargs: Optional[Dict[str, Any]] = None, **kwargs
) -> None:
self.dataset = datasets.load_dataset( self.dataset = datasets.load_dataset(
path=self.DATASET_PATH, path=self.DATASET_PATH,
name=self.DATASET_NAME, name=self.DATASET_NAME,
......
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