repo: update for new erepo function calls

iterative · May 14, 2020 · b9cef02 · b9cef02
1 parent dc71713
commit b9cef02
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/dvc/dependency/repo.py b/dvc/dependency/repo.py
@@ -56,8 +56,7 @@ def _get_checksum(self, locked=True):
             except OutputNotFoundError:
                 path = PathInfo(os.path.join(repo.root_dir, self.def_path))
                 # We are polluting our repo cache with some dir listing here
-                with repo.use_cache(self.repo.cache.local):
-                    return repo.get_checksum(path)
+                return repo.repo_tree.get_checksum(path, self.repo.cache.local)
 
     def status(self):
         current_checksum = self._get_checksum(locked=True)
@@ -79,8 +78,9 @@ def download(self, to):
             if self.def_repo.get(self.PARAM_REV_LOCK) is None:
                 self.def_repo[self.PARAM_REV_LOCK] = repo.get_rev()
 
-            with repo.use_cache(self.repo.cache.local):
-                repo.get_external(self.def_path, to.path_info)
+            repo.get_external(
+                self.def_path, to.path_info, cache=self.repo.cache.local
+            )
 
     def update(self, rev=None):
         if rev:

diff --git a/dvc/repo/fetch.py b/dvc/repo/fetch.py
@@ -78,5 +78,4 @@ def _fetch_external(self, repo_url, repo_rev, files, **kwargs):
     from dvc.external_repo import external_repo
 
     with external_repo(repo_url, repo_rev) as repo:
-        with repo.use_cache(self.cache.local):
-            return repo.fetch_external(files, **kwargs)
+        return repo.fetch_external(files, cache=self.cache.local, **kwargs)