wip

beam-cloud · Oct 31, 2024 · ef261e2 · ef261e2
1 parent e9f3192
commit ef261e2
Showing 1 changed file with 1 addition and 3 deletions.
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=False)
+@handle_error(print_traceback=True)
 @pass_channel
 def main(channel: Channel):
     function_stub: FunctionServiceStub = FunctionServiceStub(channel)
@@ -174,7 +174,6 @@ def main(channel: Channel):
     if result.exception:
         thread_pool.shutdown(wait=False)
         handle_task_failure(gateway_stub, result, task_id, container_id, container_hostname)
-        print(result.exception)
         raise result.exception
 
     # End the task and send callback
@@ -227,7 +226,6 @@ def invoke_function(
             callback_url=callback_url,
         )
     except BaseException as e:
-        print(f"{traceback.format_exc()}")
         return InvokeResult(
             result=result,
             exception=e,