5
0
justheuristic 3 жил өмнө
parent
commit
4c92d8636f

+ 2 - 1
src/client/remote_sequential.py

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