From 02715f917c775155223dcb0f9bbc5fa7b1786e7e Mon Sep 17 00:00:00 2001 From: Donny Greenberg Date: Mon, 15 Apr 2024 10:01:38 -0400 Subject: [PATCH] Catch all non-deserializable exceptions client-side (#730) --- runhouse/servers/http/http_utils.py | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/runhouse/servers/http/http_utils.py b/runhouse/servers/http/http_utils.py index dedbfda64..30baa90db 100644 --- a/runhouse/servers/http/http_utils.py +++ b/runhouse/servers/http/http_utils.py @@ -246,16 +246,13 @@ def handle_response( fn_exception = deserialize_data( response_data["data"]["error"], response_data["serialization"] ) - except RuntimeError as e: - if "RuntimeError: Failed to unpickle" in str(e): - logger.error( - f"{system_color}{err_str}: Failed to unpickle exception. Please check the logs for more " - f"information.{reset_color}" - ) - logger.error( - f"{system_color}Traceback: {fn_traceback}{reset_color}" - ) - raise e + except Exception as e: + logger.error( + f"{system_color}{err_str}: Failed to unpickle exception. Please check the logs for more " + f"information.{reset_color}" + ) + logger.error(f"{system_color}Traceback: {fn_traceback}{reset_color}") + raise e if not ( isinstance(fn_exception, StopIteration) or isinstance(fn_exception, GeneratorExit)