Skip to content
This repository was archived by the owner on Jun 30, 2022. It is now read-only.

Commit 29083f5

Browse files
committed
Use reraise_augmented for start/finish operations
----Release Notes---- [] ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=119989581
1 parent c79ccc5 commit 29083f5

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

google/cloud/dataflow/runners/common.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,14 +79,14 @@ def start(self):
7979
try:
8080
self._process_outputs(None, self.dofn.start_bundle(self.context))
8181
except BaseException as exn:
82-
raise self.augment_exception(exn)
82+
self.reraise_augmented(exn)
8383

8484
def finish(self):
8585
self.context.set_element(None)
8686
try:
8787
self._process_outputs(None, self.dofn.finish_bundle(self.context))
8888
except BaseException as exn:
89-
raise self.augment_exception(exn)
89+
self.reraise_augmented(exn)
9090

9191
def process(self, element):
9292
try:

0 commit comments

Comments
 (0)