Skip to content

Commit 3ed207a

Browse files
authored
Unify signature of Pool.acquire methods (#606)
1 parent 4fbae6f commit 3ed207a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

neo4j/io/__init__.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -829,8 +829,7 @@ def __init__(self, opener, pool_config, workspace_config, address):
829829
def __repr__(self):
830830
return "<{} address={!r}>".format(self.__class__.__name__, self.address)
831831

832-
def acquire(self, access_mode=None, timeout=None, database=None,
833-
imp_user=None, bookmarks=None):
832+
def acquire(self, access_mode=None, timeout=None, database=None, bookmarks=None):
834833
# The access_mode and database is not needed for a direct connection, its just there for consistency.
835834
return self._acquire(self.address, timeout)
836835

tests/unit/io/test_direct.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,8 @@ def opener(addr, timeout):
104104
super().__init__(opener, self.pool_config, self.workspace_config)
105105
self.address = address
106106

107-
def acquire(self, access_mode=None, timeout=None, database=None):
107+
def acquire(self, access_mode=None, timeout=None, database=None,
108+
bookmarks=None):
108109
return self._acquire(self.address, timeout)
109110

110111

0 commit comments

Comments
 (0)