fix slow test

huggingface · Nov 22, 2024 · 0ddda21 · 0ddda21
1 parent 6e2ca9b
commit 0ddda21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/got_ocr2/test_modeling_got_ocr2.py b/tests/models/got_ocr2/test_modeling_got_ocr2.py
@@ -278,7 +278,7 @@ def test_small_model_integration_test_got_ocr_format(self):
         decoded_output = self.processor.decode(
             generate_ids[0, inputs["input_ids"].shape[1] :], skip_special_tokens=True
         )
-        expected_output = "\\title{\nR'"
+        expected_output = "\\title{\nR"
         self.assertEqual(decoded_output, expected_output)
 
     @slow