style

huggingface · Feb 10, 2025 · 45a99e7 · 45a99e7
1 parent e8e0d34
commit 45a99e7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/open_r1/rewards.py b/src/open_r1/rewards.py
@@ -152,9 +152,9 @@ def cosine_scaled_reward(completions, solution, **kwargs):
 
 def get_repetition_penalty_reward(ngram_size: int, max_penalty: float):
     """
-    Computes N-gram repetition penalty as described in Appendix C.2 of https://arxiv.org/abs/2502.03373. 
+    Computes N-gram repetition penalty as described in Appendix C.2 of https://arxiv.org/abs/2502.03373.
     Reference implementation from: https://github.com/eddycmu/demystify-long-cot/blob/release/openrlhf/openrlhf/reward/repetition.py
-    
+
     Args:
     ngram_size: size of the n-grams
     max_penalty: Maximum (negative) penalty for wrong answers