Prevent double close in HTTP client __del__

triton-inference-server · Feb 6, 2024 · 0f476ad · 0f476ad
1 parent 1dc9836
commit 0f476ad
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/python/library/tritonclient/http/_client.py b/src/python/library/tritonclient/http/_client.py
@@ -190,6 +190,7 @@ def __init__(
         )
         self._pool = gevent.pool.Pool(max_greenlets)
         self._verbose = verbose
+        self._closed = False
 
     def __enter__(self):
         return self
@@ -205,8 +206,10 @@ def close(self):
         will result in an Error.
 
         """
-        self._pool.join()
-        self._client_stub.close()
+        if not self._closed:
+            self._pool.join()
+            self._client_stub.close()
+            self._closed = True
 
     def _get(self, request_uri, headers, query_params):
         """Issues the GET request to the server