Skip to content

GitLab

"mmdet3d/vscode:/vscode.git/clone" did not exist on "318499acd0c05fe1fee638d22186067a92a0684f"
Commit 3d7f777d authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

make process_docs boolean more clear

parent 380c62f9
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
lm_eval/api/task.py
View file @ 3d7f777d
...@@ -665,7 +665,7 @@ class ConfigurableTask(Task): ...@@ -665,7 +665,7 @@ class ConfigurableTask(Task):
def training_docs(self): def training_docs(self):
if self.has_training_docs(): if self.has_training_docs():
if self._config.process_docs: if self._config.process_docs is not None:
return self._config.process_docs( return self._config.process_docs(
self.dataset[self._config.training_split] self.dataset[self._config.training_split]
) )
...@@ -673,7 +673,7 @@ class ConfigurableTask(Task): ...@@ -673,7 +673,7 @@ class ConfigurableTask(Task):
def validation_docs(self): def validation_docs(self):
if self.has_validation_docs(): if self.has_validation_docs():
if self._config.process_docs: if self._config.process_docs is not None:
return self._config.process_docs( return self._config.process_docs(
self.dataset[self._config.validation_split] self.dataset[self._config.validation_split]
) )
...@@ -681,7 +681,7 @@ class ConfigurableTask(Task): ...@@ -681,7 +681,7 @@ class ConfigurableTask(Task):
def test_docs(self): def test_docs(self):
if self.has_test_docs(): if self.has_test_docs():
if self._config.process_docs: if self._config.process_docs is not None:
return self._config.process_docs(self.dataset[self._config.test_split]) return self._config.process_docs(self.dataset[self._config.test_split])
return self.dataset[self._config.test_split] return self.dataset[self._config.test_split]
......
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