precommits

model/supervised_finetuning/custom_datasets/__init__.py

@@ -30,6 +30,7 @@ summarization_config_mapping = {
 QA_DATASETS = ["squad_v2", "adversarial_qa", "trivia_qa_context", "trivia_qa_noconext"]
 SUMMARIZATION_DATASETS = ["xsum", "cnn_dailymail", "samsum", "multi_news"]
 
+
 def index_squad_v2(example):
     return example["title"] + ". " + example["context"] + " " + example["question"], example["answers"]["text"][0]
 
@@ -159,4 +160,4 @@ def get_one_dataset(conf, dataset_name):
     else:
         raise ValueError(f"Unknown dataset {dataset_name}")
 
-    return train, eval
+    return train, eval