Update pool.py

ydb-platform · Sep 20, 2023 · 6709a63 · 6709a63
1 parent fb563b7
commit 6709a63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ydb/aio/pool.py b/ydb/aio/pool.py
@@ -34,7 +34,7 @@ async def get(self, preferred_endpoint=None, fast_fail=False, wait_timeout=10):
             return self.connections_by_node_id[preferred_endpoint.node_id]
 
         if preferred_endpoint is not None and preferred_endpoint.endpoint in self.connections:
-            return self.connections[preferred_endpoint]
+            return self.connections[preferred_endpoint.endpoint]
 
         for conn_lst in self.conn_lst_order:
             try: