From 1f90b426fb2217974b3cf3d1c3f05b3e950e772c Mon Sep 17 00:00:00 2001 From: Alexander Schepanovski Date: Sun, 21 Jul 2019 13:41:04 +0700 Subject: [PATCH] remote: prettify get_connection/pool() signatures --- dvc/remote/pool.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/dvc/remote/pool.py b/dvc/remote/pool.py index e15fdb689..3f92056db 100644 --- a/dvc/remote/pool.py +++ b/dvc/remote/pool.py @@ -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():