nit

facebookresearch · Sep 26, 2024 · 2a24bcf · 2a24bcf
1 parent a5eb8da
commit 2a24bcf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/projects/self_taught_evaluator/src/prepare_dpo_data.py b/projects/self_taught_evaluator/src/prepare_dpo_data.py
@@ -54,6 +54,7 @@ def get_input_key(response):
     input_key = f"<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n[User Question]{eval_example}"
     return system_prompt, input_key
 
+
 def prepare_dpo_data(
     generation_dir: str,
     output_dir: str,

diff --git a/projects/self_taught_evaluator/src/prepare_sft_data.py b/projects/self_taught_evaluator/src/prepare_sft_data.py
@@ -16,6 +16,7 @@
 
 random.seed(42)
 
+
 def rejection_sampling(outputs, true_label, majority_voting=None):
     random.shuffle(outputs)
     positives = []