Show name of experiments in the queue (#6382)

* Show name of stashed experiments

* Add tests for it

* Reorder the stash and commit rev name
This commit is contained in:
Gao
2021-08-06 11:02:16 +09:00
committed by GitHub
parent 6c33fbb617
commit 3d604a8455
3 changed files with 11 additions and 6 deletions
+5 -1
View File
@@ -879,7 +879,11 @@ class Experiments:
exclude = f"{EXEC_NAMESPACE}/*"
ref = self.scm.describe(rev, base=EXPS_NAMESPACE, exclude=exclude)
if ref:
return ExpRefInfo.from_ref(ref).name
name = ExpRefInfo.from_ref(ref).name
if name:
return name
if rev in self.stash_revs:
return self.stash_revs[rev].name
return None
def get_running_exps(self) -> Dict[str, int]:
+2 -4
View File
@@ -61,10 +61,7 @@ def _collect_experiment_commit(
if name:
break
if not name:
if stash:
pass
else:
name = repo.experiments.get_exact_name(rev)
name = repo.experiments.get_exact_name(rev)
if name:
name = name.rsplit("/")[-1]
res["name"] = name
@@ -187,6 +184,7 @@ def show(
experiment = _collect_experiment_commit(
repo,
stash_rev,
sha_only=sha_only,
stash=stash_rev not in running,
param_deps=param_deps,
running=running,
+4 -1
View File
@@ -68,12 +68,15 @@ def test_show_experiment(tmp_dir, scm, dvc, exp_stage, workspace):
def test_show_queued(tmp_dir, scm, dvc, exp_stage):
baseline_rev = scm.get_rev()
dvc.experiments.run(exp_stage.addressing, params=["foo=2"], queue=True)
dvc.experiments.run(
exp_stage.addressing, params=["foo=2"], queue=True, name="test_name"
)
exp_rev = dvc.experiments.scm.resolve_rev(f"{EXPS_STASH}@{{0}}")
results = dvc.experiments.show()[baseline_rev]
assert len(results) == 2
exp = results[exp_rev]["data"]
assert exp["name"] == "test_name"
assert exp["queued"]
assert exp["params"]["params.yaml"] == {"data": {"foo": 2}}