gemini-feedback

Author: TJ5

tests/sampling/test_text.py

@@ -221,7 +221,7 @@ def test_short_prompt_request(self):
         result = self.sampler.sample(scenario)
         self.assertIsInstance(result, UserChatRequest)
         # The prompt will be the 4-digit number, truncated to 1 char
-        self.assertEqual(result.prompt, str(result.prompt)[0])
+        self.assertEqual(len(result.prompt), 1)
         self.assertGreater(len(result.prompt), 0)
 
     def test_empty_dataset(self):