Archives/petals
2
0
Fork 0
mirror of https://github.com/bigscience-workshop/petals synced 2024-11-18 03:25:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

forward exc details

Browse Source
This commit is contained in:
justheuristic 2022-06-29 15:31:11 +03:00
parent e7f716502c
commit 4c92d8636f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/client/remote_sequential.py
View File

@ -121,11 +121,12 @@ class RemoteSequentialInferenceSession:
def close(self, *exc_details):
"""Finish a given inference session, close the underlying connection"""
assert not self.closed
self.stack.__exit__(*exc_details)
self.active_sessions.clear()
self.closed = True
def __exit__(self, *exc_details):
self.close()
self.close(*exc_details)
def __del__(self):
self.close()