diff --git a/src/petals/client/inference_session.py b/src/petals/client/inference_session.py index 5472d68ae..376291a45 100644 --- a/src/petals/client/inference_session.py +++ b/src/petals/client/inference_session.py @@ -266,7 +266,7 @@ def _enter_server_sessions(self, chosen_spans: List[RemoteSpanInfo]) -> List[_Se server_sessions.append(session) session.__enter__() return server_sessions - except: + except Exception: self._exit_server_sessions(server_sessions) raise diff --git a/src/petals/server/server.py b/src/petals/server/server.py index 45884e3b7..b90033cb5 100644 --- a/src/petals/server/server.py +++ b/src/petals/server/server.py @@ -534,7 +534,7 @@ def create( if should_validate_reachability: validate_reachability(dht.peer_id) - except: + except Exception: logger.debug("Shutting down backends") for backend in blocks.values(): backend.shutdown()