apply lint

Author: Ktakuya332C

flexeval/core/multiple_choice_dataset/template_based.py

@@ -77,8 +77,8 @@ def __getitem__(self, i: int) -> MultipleChoiceInstance:
             choices = [" " + c for c in choices]
 
         answer_index = int(self.answer_index_template.render(**item))
-        if not (0 <= answer_index and answer_index < len(choices)):
-            msg = f"at least {answer_idx+1} choices required, but got {choices}"
+        if not (answer_index >= 0 and answer_index < len(choices)):
+            msg = f"at least {answer_index+1} choices required, but got {choices}"
             raise ValueError(msg)
 
         return MultipleChoiceInstance(

tests/core/multiple_choice_dataset/test_template_based.py

@@ -73,6 +73,7 @@ def test_template_multiple_choice_dataset(
         "test_additional_input": "additional: Who wrote 'Romeo and Juliet'?",
     }
 
+
 @pytest.mark.parametrize(
     ("dataset_class", "kwargs"),
     DATASETS_TO_TEST,