Catch all non-deserializable exceptions client-side (#730)

run-house · Apr 15, 2024 · 02715f9 · 02715f9
1 parent 8a726aa
commit 02715f9
Showing 1 changed file with 7 additions and 10 deletions.
diff --git 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)