From 8620b6e5afb6b6878d34d2e2b4109bbca93ec60c Mon Sep 17 00:00:00 2001 From: David de la Iglesia Castro Date: Tue, 2 May 2023 16:31:17 +0200 Subject: [PATCH] exp: Update `--rev` to support `append` action. Closes #9390 --- dvc/commands/experiments/__init__.py | 1 + tests/unit/command/test_experiments.py | 10 +++++----- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/dvc/commands/experiments/__init__.py b/dvc/commands/experiments/__init__.py index d7e77d81e..4bbd79063 100644 --- a/dvc/commands/experiments/__init__.py +++ b/dvc/commands/experiments/__init__.py @@ -82,6 +82,7 @@ def add_rev_selection_flags( experiments_subcmd_parser.add_argument( "--rev", type=str, + action="append", default=None, help=msg, metavar="", diff --git a/tests/unit/command/test_experiments.py b/tests/unit/command/test_experiments.py index 5756538fe..a08a45286 100644 --- a/tests/unit/command/test_experiments.py +++ b/tests/unit/command/test_experiments.py @@ -116,7 +116,7 @@ def test_experiments_show(dvc, scm, mocker): hide_queued=True, hide_failed=True, num=1, - revs="foo", + revs=["foo"], sha_only=True, param_deps=True, fetch_running=True, @@ -226,7 +226,7 @@ def test_experiments_list(dvc, scm, mocker): m.assert_called_once_with( cmd.repo, git_remote="origin", - rev="foo", + rev=["foo"], all_commits=True, num=-1, ) @@ -265,7 +265,7 @@ def test_experiments_push(dvc, scm, mocker): cmd.repo, "origin", ["experiment1", "experiment2"], - rev="foo", + rev=["foo"], all_commits=True, num=2, force=True, @@ -322,7 +322,7 @@ def test_experiments_pull(dvc, scm, mocker): cmd.repo, "origin", ["experiment"], - rev="foo", + rev=["foo"], all_commits=True, num=1, force=True, @@ -371,7 +371,7 @@ def test_experiments_remove_flag(dvc, scm, mocker, capsys, caplog): cmd.repo, exp_names=[], all_commits=True, - rev="foo", + rev=["foo"], num=2, queue=False, git_remote="myremote",