Additional context for decoding errors.

apache · Jun 6, 2023 · 1546cf2 · 1546cf2
1 parent 98210d9
commit 1546cf2
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
@@ -226,8 +226,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):