pyink

Author: hengtaoguo

tests/unit/rl_utils_test.py

@@ -180,10 +180,11 @@ def test_score_duplicate_reasoning_start_tag(self):
   @pytest.mark.cpu_only
   def test_score_multiple_completions(self):
     # Multiple completions at once -> one score per entry
-    scores = self._score([
+    multi_completions = [
         "<reasoning>think</reasoning><answer>42</answer>",  # 2.0
         "no tags here",  # -2.0
-    ])
+    ]
+    scores = self._score(multi_completions)
     self.assertEqual(len(scores), 2)
     self.assertEqual(scores[0], 2.0)
     self.assertEqual(scores[1], -2.0)