diff --git a/dvc/remote/base.py b/dvc/remote/base.py index 90fc167e1..811fd9fe5 100644 --- a/dvc/remote/base.py +++ b/dvc/remote/base.py @@ -75,11 +75,9 @@ class DirCacheError(DvcException): def index_locked(f): @wraps(f) def wrapper(remote_obj, *args, **kwargs): - remote = kwargs.get("remote") - if remote: - with remote.index: - return f(remote_obj, *args, **kwargs) - return f(remote_obj, *args, **kwargs) + remote = kwargs.get("remote", remote_obj) + with remote.index: + return f(remote_obj, *args, **kwargs) return wrapper