Additional context for decoding errors. (#27024)

apache · Oct 9, 2023 · f63e9ed · f63e9ed
1 parent 21f822f
commit f63e9ed
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/runners/worker/bundle_processor.py b/sdks/python/apache_beam/runners/worker/bundle_processor.py
@@ -227,8 +227,13 @@ def process_encoded(self, encoded_windowed_values):
         if self.index == self.stop - 1:
           return
         self.index += 1
-      decoded_value = self.windowed_coder_impl.decode_from_stream(
-          input_stream, True)
+      try:
+        decoded_value = self.windowed_coder_impl.decode_from_stream(
+            input_stream, True)
+      except Exception as exn:
+        raise ValueError(
+            "Error decoding input stream with coder " +
+            self.windowed_coder) from exn
       self.output(decoded_value)
 
   def monitoring_infos(self, transform_id, tag_to_pcollection_id):