From 4ee4f137f2c6303a499eafe1e4287fc8dd93241b Mon Sep 17 00:00:00 2001 From: Peter Rowlands Date: Mon, 26 Jun 2023 13:47:34 +0900 Subject: [PATCH] exp pull: update for index fetch --- dvc/repo/experiments/executor/ssh.py | 3 +-- dvc/repo/experiments/pull.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/dvc/repo/experiments/executor/ssh.py b/dvc/repo/experiments/executor/ssh.py index 85431dbea..6ec153238 100644 --- a/dvc/repo/experiments/executor/ssh.py +++ b/dvc/repo/experiments/executor/ssh.py @@ -210,8 +210,7 @@ class SSHExecutor(BaseExecutor): """Collect DVC cache.""" from dvc.repo.experiments.pull import _pull_cache - with self.get_odb() as odb: - _pull_cache(repo, exp_ref, run_cache=run_cache, odb=odb) + _pull_cache(repo, exp_ref, run_cache=run_cache) @contextmanager def get_odb(self): diff --git a/dvc/repo/experiments/pull.py b/dvc/repo/experiments/pull.py index 6841cfd4c..349e0ffa6 100644 --- a/dvc/repo/experiments/pull.py +++ b/dvc/repo/experiments/pull.py @@ -107,10 +107,9 @@ def _pull_cache( dvc_remote=None, jobs=None, run_cache=False, - odb=None, ): if isinstance(refs, ExpRefInfo): refs = [refs] revs = list(exp_commits(repo.scm, refs)) logger.debug("dvc fetch experiment '%s'", refs) - repo.fetch(jobs=jobs, remote=dvc_remote, run_cache=run_cache, revs=revs, odb=odb) + repo.fetch(jobs=jobs, remote=dvc_remote, run_cache=run_cache, revs=revs)