Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-11 07:18:28 +01:00
Code Issues Packages Projects Releases Wiki Activity

remote: prettify get_connection/pool() signatures

Browse Source
This commit is contained in:
Alexander Schepanovski
2019-07-21 13:41:04 +07:00
parent 6afc2a38ba
commit 1f90b426fb
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dvc/remote/pool.py
+4 -4
View File
@@ -4,8 +4,8 @@ from funcy import memoize
@contextmanager
def get_connection(*args, **kwargs):
pool = get_pool(*args, **kwargs)
def get_connection(conn_func, *args, **kwargs):
pool = get_pool(conn_func, *args, **kwargs)
conn = pool.get_connection()
try:
yield conn
@@ -17,8 +17,8 @@ def get_connection(*args, **kwargs):
@memoize
def get_pool(*args, **kwargs):
return Pool(*args, **kwargs)
def get_pool(conn_func, *args, **kwargs):
return Pool(conn_func, *args, **kwargs)
def close_pools():