exp list: fix ui for tags/branches

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
+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):
+1 -1
View File
@@ -165,7 +165,7 @@ def test_list_remote(tmp_dir, scm, dvc, git_downstream, exp_stage, use_url):
exp_list = downstream_exp.ls(all_commits=True, git_remote=remote)
assert {key: set(val) for key, val in exp_list.items()} == {
baseline_a: {(ref_info_a.name, None), (ref_info_b.name, None)},
"master": {(ref_info_c.name, None)},
"refs/heads/master": {(ref_info_c.name, None)},
}
+1 -1
View File
@@ -71,7 +71,7 @@ def test_exp_save_after_commit(tmp_dir, dvc, scm):
all_exps = dvc.experiments.ls(all_commits=True)
assert all_exps[baseline][0][0] == "exp-1"
assert all_exps["master"][0][0] == "exp-2"
assert all_exps["refs/heads/master"][0][0] == "exp-2"
def test_exp_save_with_staged_changes(tmp_dir, dvc, scm):
+1 -1
View File
@@ -198,7 +198,7 @@ def test_experiments_list_format(mocker, capsys, args, expected, dvc, scm):
mocker.patch(
"dvc.repo.experiments.ls.ls",
return_value={
"main": [
"refs/heads/main": [
("exp-a", "sha-a"),
]
},