indent update

d01032d2 · Konrad · a4bc4846 · d01032d2
Commit d01032d2 authored May 14, 2024 by Konrad
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 11 deletions

lm_eval/api/task.py lm_eval/api/task.py +10 -11

No files found.
--- a/lm_eval/api/task.py
+++ b/lm_eval/api/task.py
@@ -1077,17 +1077,16 @@ class ConfigurableTask(Task):
        else:
            if self.multiple_input:
                return labeled_examples
-            else:
+            if isinstance(example, str):
-                if isinstance(example, str):
+                return labeled_examples + example
-                    return labeled_examples + example
+            elif isinstance(example, list):
-                elif isinstance(example, list):
+                return [labeled_examples + ex for ex in example]
-                    return [labeled_examples + ex for ex in example]
+            elif isinstance(example, int):
-                elif isinstance(example, int):
+                if self.config.doc_to_choice is not None:
-                    if self.config.doc_to_choice is not None:
+                    choices = self.doc_to_choice(doc)
-                        choices = self.doc_to_choice(doc)
+                    return labeled_examples + choices[example]
-                        return labeled_examples + choices[example]
+                else:
-                    else:
+                    return labeled_examples + str(example)
-                        return labeled_examples + str(example)
    def apply_filters(self):
        """Iterates over FilterEnsembles and applies them to instances"""