exp pull: update for index fetch

This commit is contained in:
Peter Rowlands
2023-06-26 14:09:24 +09:00
committed by Peter Rowlands (변기호)
parent a750a75016
commit 4ee4f137f2
2 changed files with 2 additions and 4 deletions
+1 -2
View File
@@ -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):
+1 -2
View File
@@ -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)