wip

beam-cloud · Oct 31, 2024 · 0ab00be · 0ab00be
1 parent ef261e2
commit 0ab00be
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/sdk/src/beta9/channel.py b/sdk/src/beta9/channel.py
@@ -186,7 +186,7 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:
 
 
 @contextmanager
-def handle_error(print_traceback: bool = False):
+def handle_error():
     exit_code = 0
     try:
         yield
@@ -201,8 +201,7 @@ def handle_error(print_traceback: bool = False):
         exit_code = 1
     finally:
         if exit_code != 0:
-            if print_traceback:
-                print(traceback.format_exc())
+            print(traceback.format_exc())
             sys.exit(exit_code)
 
 

diff --git a/sdk/src/beta9/runner/function.py b/sdk/src/beta9/runner/function.py
@@ -136,7 +136,7 @@ def _monitor_task(
 
 
 @json_output_interceptor(task_id=config.task_id)
-@handle_error(print_traceback=True)
+@handle_error()
 @pass_channel
 def main(channel: Channel):
     function_stub: FunctionServiceStub = FunctionServiceStub(channel)