mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-14 07:18:27 +01:00
4cc06339c8
* Clean up remotes's exps fix #6006 1. add a new argument `--git-remote` to `dvc exp remove` 2. support remote a special remote exp 3. add tests for it 4. fix #6421 5. add a test for #6421. 6. extract some functions from the `dvc pull`, `dvc push`, `dvc remove` to utils. 7. add a tests for this new util function `resolve_exp_ref` 8. add `__init__.py` to some test package for the pytest fail Co-authored-by: Jorge Orpinel <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Dave Berenbaum <[email protected]>
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
import logging
|
|
|
|
from dvc.exceptions import DvcException, InvalidArgumentError
|
|
from dvc.repo import locked
|
|
from dvc.repo.scm_context import scm_context
|
|
|
|
from .utils import exp_commits, resolve_exp_ref
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@locked
|
|
@scm_context
|
|
def pull(
|
|
repo, git_remote, exp_name, *args, force=False, pull_cache=False, **kwargs
|
|
):
|
|
exp_ref = resolve_exp_ref(repo.scm, exp_name, git_remote)
|
|
if not exp_ref:
|
|
raise InvalidArgumentError(
|
|
f"Experiment '{exp_name}' does not exist in '{git_remote}'"
|
|
)
|
|
|
|
def on_diverged(refname: str, rev: str) -> bool:
|
|
if repo.scm.get_ref(refname) == rev:
|
|
return True
|
|
raise DvcException(
|
|
f"Local experiment '{exp_name}' has diverged from remote "
|
|
"experiment with the same name. To override the local experiment "
|
|
"re-run with '--force'."
|
|
)
|
|
|
|
refspec = f"{exp_ref}:{exp_ref}"
|
|
logger.debug("git pull experiment '%s' -> '%s'", git_remote, refspec)
|
|
repo.scm.fetch_refspecs(
|
|
git_remote, [refspec], force=force, on_diverged=on_diverged
|
|
)
|
|
|
|
if pull_cache:
|
|
_pull_cache(repo, exp_ref, **kwargs)
|
|
|
|
|
|
def _pull_cache(repo, exp_ref, dvc_remote=None, jobs=None, run_cache=False):
|
|
revs = list(exp_commits(repo.scm, [exp_ref]))
|
|
logger.debug("dvc fetch experiment '%s'", exp_ref)
|
|
repo.fetch(jobs=jobs, remote=dvc_remote, run_cache=run_cache, revs=revs)
|