mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
@@ -8,7 +8,7 @@ from PyInstaller.utils.hooks import collect_submodules, is_module_or_submodule
|
|||||||
_EXCLUDES = ("celery.bin", "celery.contrib")
|
_EXCLUDES = ("celery.bin", "celery.contrib")
|
||||||
hiddenimports = collect_submodules(
|
hiddenimports = collect_submodules(
|
||||||
"celery",
|
"celery",
|
||||||
filter=lambda name: not any(
|
filter=lambda name: (
|
||||||
is_module_or_submodule(name, module) for module in _EXCLUDES
|
not any(is_module_or_submodule(name, module) for module in _EXCLUDES)
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -188,7 +188,7 @@ class Experiments:
|
|||||||
rev_names = self.get_exact_name(revs)
|
rev_names = self.get_exact_name(revs)
|
||||||
for rev in revs:
|
for rev in revs:
|
||||||
name = rev_names[rev]
|
name = rev_names[rev]
|
||||||
names.append(name if name else rev[:7])
|
names.append(name or rev[:7])
|
||||||
ui.write("\nRan experiment(s): {}".format(", ".join(names)))
|
ui.write("\nRan experiment(s): {}".format(", ".join(names)))
|
||||||
if tmp_dir:
|
if tmp_dir:
|
||||||
ui.write(
|
ui.write(
|
||||||
|
|||||||
@@ -328,7 +328,7 @@ def collect(
|
|||||||
name=baseline_names.get(baseline_rev),
|
name=baseline_names.get(baseline_rev),
|
||||||
data=baseline_data.data,
|
data=baseline_data.data,
|
||||||
error=baseline_data.error,
|
error=baseline_data.error,
|
||||||
experiments=experiments if experiments else None,
|
experiments=experiments or None,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
return result
|
return result
|
||||||
|
|||||||
@@ -757,7 +757,7 @@ class BaseExecutor(ABC):
|
|||||||
logger.debug("Commit to current experiment branch '%s'", branch)
|
logger.debug("Commit to current experiment branch '%s'", branch)
|
||||||
else:
|
else:
|
||||||
baseline_rev = scm.get_ref(EXEC_BASELINE)
|
baseline_rev = scm.get_ref(EXEC_BASELINE)
|
||||||
name = exp_name if exp_name else f"exp-{exp_hash[:5]}"
|
name = exp_name or f"exp-{exp_hash[:5]}"
|
||||||
ref_info = ExpRefInfo(baseline_rev, name)
|
ref_info = ExpRefInfo(baseline_rev, name)
|
||||||
branch = str(ref_info)
|
branch = str(ref_info)
|
||||||
old_ref = None
|
old_ref = None
|
||||||
|
|||||||
@@ -34,7 +34,7 @@ class ExpRefInfo:
|
|||||||
|
|
||||||
def __init__(self, baseline_sha: str, name: Optional[str] = None):
|
def __init__(self, baseline_sha: str, name: Optional[str] = None):
|
||||||
self.baseline_sha = baseline_sha
|
self.baseline_sha = baseline_sha
|
||||||
self.name: str = name if name else ""
|
self.name: str = name or ""
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return "/".join(self.parts)
|
return "/".join(self.parts)
|
||||||
|
|||||||
@@ -67,7 +67,7 @@ class ExpStash(Stash):
|
|||||||
branch: Optional[str] = None,
|
branch: Optional[str] = None,
|
||||||
) -> str:
|
) -> str:
|
||||||
msg = cls.MESSAGE_FORMAT.format(
|
msg = cls.MESSAGE_FORMAT.format(
|
||||||
rev=rev, baseline_rev=baseline_rev, name=name if name else ""
|
rev=rev, baseline_rev=baseline_rev, name=name or ""
|
||||||
)
|
)
|
||||||
branch_msg = f":{branch}" if branch else ""
|
branch_msg = f":{branch}" if branch else ""
|
||||||
return f"{msg}{branch_msg}"
|
return f"{msg}{branch_msg}"
|
||||||
@@ -124,9 +124,7 @@ class ApplyStash(Stash):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def format_message(cls, head_rev: str, rev: str, name: Optional[str] = None) -> str:
|
def format_message(cls, head_rev: str, rev: str, name: Optional[str] = None) -> str:
|
||||||
return cls.MESSAGE_FORMAT.format(
|
return cls.MESSAGE_FORMAT.format(head_rev=head_rev, rev=rev, name=name or "")
|
||||||
head_rev=head_rev, rev=rev, name=name if name else ""
|
|
||||||
)
|
|
||||||
|
|
||||||
@contextmanager
|
@contextmanager
|
||||||
def preserve_workspace(
|
def preserve_workspace(
|
||||||
|
|||||||
@@ -200,7 +200,7 @@ def exp_commits(
|
|||||||
) -> Iterable[str]:
|
) -> Iterable[str]:
|
||||||
"""Iterate over all experiment commits."""
|
"""Iterate over all experiment commits."""
|
||||||
shas: set[str] = set()
|
shas: set[str] = set()
|
||||||
refs = ref_infos if ref_infos else exp_refs(scm)
|
refs = ref_infos or exp_refs(scm)
|
||||||
for ref_info in refs:
|
for ref_info in refs:
|
||||||
shas.update(scm.branch_revs(str(ref_info), ref_info.baseline_sha))
|
shas.update(scm.branch_revs(str(ref_info), ref_info.baseline_sha))
|
||||||
yield from shas
|
yield from shas
|
||||||
|
|||||||
@@ -53,7 +53,7 @@ def test_new_simple(tmp_dir, scm, dvc, exp_stage, mocker, name, workspace):
|
|||||||
if workspace:
|
if workspace:
|
||||||
assert (tmp_dir / "metrics.yaml").read_text().strip() == "foo: 2"
|
assert (tmp_dir / "metrics.yaml").read_text().strip() == "foo: 2"
|
||||||
|
|
||||||
exp_name = name if name else ref_info.name
|
exp_name = name or ref_info.name
|
||||||
assert dvc.experiments.get_exact_name([exp])[exp] == exp_name
|
assert dvc.experiments.get_exact_name([exp])[exp] == exp_name
|
||||||
assert resolve_rev(scm, exp_name) == exp
|
assert resolve_rev(scm, exp_name) == exp
|
||||||
|
|
||||||
|
|||||||
@@ -28,7 +28,7 @@ def test_exp_save(tmp_dir, dvc, scm, name):
|
|||||||
assert ref_info
|
assert ref_info
|
||||||
assert ref_info.baseline_sha == baseline
|
assert ref_info.baseline_sha == baseline
|
||||||
|
|
||||||
exp_name = name if name else ref_info.name
|
exp_name = name or ref_info.name
|
||||||
assert dvc.experiments.get_exact_name([exp])[exp] == exp_name
|
assert dvc.experiments.get_exact_name([exp])[exp] == exp_name
|
||||||
assert resolve_rev(scm, exp_name) == exp
|
assert resolve_rev(scm, exp_name) == exp
|
||||||
|
|
||||||
|
|||||||
@@ -196,7 +196,7 @@ def test_diff_show_markdown_and_hash(mocker, show_hash, dvc):
|
|||||||
cmd = args.func(args)
|
cmd = args.func(args)
|
||||||
|
|
||||||
diff = {}
|
diff = {}
|
||||||
show_hash = show_hash if show_hash else False
|
show_hash = show_hash or False
|
||||||
mock_show_markdown = mocker.patch("dvc.commands.diff._show_markdown")
|
mock_show_markdown = mocker.patch("dvc.commands.diff._show_markdown")
|
||||||
mocker.patch("dvc.repo.Repo.diff", return_value=diff.copy())
|
mocker.patch("dvc.repo.Repo.diff", return_value=diff.copy())
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user