From 548c75bc7c279fdbf46ca0b8a6201d2abdbd7644 Mon Sep 17 00:00:00 2001 From: Ruslan Kuprieiev Date: Tue, 14 May 2024 17:52:57 +0300 Subject: [PATCH] fix(agent): catch BaseException when deserializing user function --- src/isolate/connections/grpc/agent.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/isolate/connections/grpc/agent.py b/src/isolate/connections/grpc/agent.py index 77b59de..b0c59fc 100644 --- a/src/isolate/connections/grpc/agent.py +++ b/src/isolate/connections/grpc/agent.py @@ -107,10 +107,10 @@ def execute_function( ) try: - # TODO: technically any sort of exception could be raised here, since + # NOTE: technically any sort of exception could be raised here, since # depickling is basically involves code execution from the *user*. function = from_grpc(function) - except SerializationError as exc: + except BaseException as exc: return exc, True, traceback.format_exc() if not callable(function):