fix doc_to_image

90ba03ac · lintangsutawika · 9692aa05 · 90ba03ac
Commit 90ba03ac authored Jul 10, 2024 by lintangsutawika
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/api/task.py lm_eval/api/task.py +1 -1

No files found.
--- a/lm_eval/api/task.py
+++ b/lm_eval/api/task.py
@@ -1280,7 +1280,7 @@ class ConfigurableTask(Task):

    def doc_to_image(self, doc: Any) -> Union[int, str, list]:
        if self.config.doc_to_image is None:
-            eval_logger.error("doc_to_image was called but not set in config")
+            return None
        else:
            doc_to_image = self.config.doc_to_image