[Testing Github workflow] - Fixing test_zero_max_penalty_returns_zero…

…() test
huggingface · Feb 10, 2025 · d1a4c65 · d1a4c65
1 parent 2515a0a
commit d1a4c65
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_rewards.py b/tests/test_rewards.py
@@ -120,7 +120,9 @@ def test_positive_max_penalty_raises_value_error(self):
 
     def test_zero_max_penalty_returns_zero(self):
         reward_fn = get_repetition_penalty_reward(ngram_size=2, max_penalty=0.0)
-        self.assertEqual(reward_fn, 0)
+        completions = [[{"content": "test test test"}]]
+        rewards = reward_fn(completions)
+        self.assertEqual(rewards, [0.0])
 
     def test_no_repetition(self):
         reward_fn = get_repetition_penalty_reward(ngram_size=2, max_penalty=-1.0)