Fix

Author: merrymercy

test/srt/test_vision_chunked_prefill.py

@@ -178,7 +178,7 @@ def _test_chunked_prefill(self, batches, num_frames):
             print(output_chunked)
             print("output without chunked prefill:")
             print(output_no_chunked)
-            assert output_chunked == output_no_chunked
+            self.assertEqual(output_chunked, output_no_chunked)
 
     def test_chunked_prefill(self):
         self._test_chunked_prefill(batches=[False, True], num_frames=[1, [2, 6, 8, 10]])