Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-02 07:18:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

exp list: fix ui for tags/branches

Browse Source
This commit is contained in:
dberenbaum
2023-06-20 15:52:18 +09:00
committed by Peter Rowlands (변기호)
parent 1835c8c6e9
commit bf2f08da09
6 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/func/experiments/test_experiments.py
+5 -5
View File
@@ -362,7 +362,7 @@ def test_list(tmp_dir, scm, dvc, exp_stage):
exp_c = first(results)
ref_info_c = first(exp_refs_by_rev(scm, exp_c))
assert dvc.experiments.ls() == {"master": [(ref_info_c.name, exp_c)]}
assert dvc.experiments.ls() == {"refs/heads/master": [(ref_info_c.name, exp_c)]}
exp_list = dvc.experiments.ls(rev=ref_info_a.baseline_sha)
assert {key: set(val) for key, val in exp_list.items()} == {
@@ -372,20 +372,20 @@ def test_list(tmp_dir, scm, dvc, exp_stage):
exp_list = dvc.experiments.ls(rev=[baseline_a, scm.get_rev()])
assert {key: set(val) for key, val in exp_list.items()} == {
baseline_a: {(ref_info_a.name, exp_a), (ref_info_b.name, exp_b)},
"master": {(ref_info_c.name, exp_c)},
"refs/heads/master": {(ref_info_c.name, exp_c)},
}
exp_list = dvc.experiments.ls(all_commits=True)
assert {key: set(val) for key, val in exp_list.items()} == {
baseline_a: {(ref_info_a.name, exp_a), (ref_info_b.name, exp_b)},
"master": {(ref_info_c.name, exp_c)},
"refs/heads/master": {(ref_info_c.name, exp_c)},
}
scm.checkout("branch", True)
exp_list = dvc.experiments.ls(all_commits=True)
assert {key: set(val) for key, val in exp_list.items()} == {
baseline_a: {(ref_info_a.name, exp_a), (ref_info_b.name, exp_b)},
"branch": {(ref_info_c.name, exp_c)},
"refs/heads/branch": {(ref_info_c.name, exp_c)},
}
@@ -676,7 +676,7 @@ def test_experiment_unchanged(tmp_dir, scm, dvc, exp_stage):
dvc.experiments.run(exp_stage.addressing)
dvc.experiments.run(exp_stage.addressing)
assert len(dvc.experiments.ls()["master"]) == 2
assert len(dvc.experiments.ls()["refs/heads/master"]) == 2
def test_experiment_run_dry(tmp_dir, scm, dvc, exp_stage):