From 54539d0db1d8e74df95aa631f3774d2c8b65ae79 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sat, 30 Nov 2024 21:29:30 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- tests/func/experiments/test_remove.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/tests/func/experiments/test_remove.py b/tests/func/experiments/test_remove.py index 4abc4b94fc..aa019d07c8 100644 --- a/tests/func/experiments/test_remove.py +++ b/tests/func/experiments/test_remove.py @@ -42,12 +42,17 @@ def test_remove_all_queued_experiments(tmp_dir, scm, dvc, exp_stage): assert len(dvc.experiments.stash_revs) == 0 assert scm.get_ref(str(ref_info)) is not None + def test_remove_all_experiments_queued_and_commits(tmp_dir, scm, dvc, exp_stage): queue_length = 3 for i in range(queue_length): - dvc.experiments.run(exp_stage.addressing, params=[f"foo={i}"], name=f"exp{i}", queue=True) + dvc.experiments.run( + exp_stage.addressing, params=[f"foo={i}"], name=f"exp{i}", queue=True + ) - results = dvc.experiments.run(exp_stage.addressing, params=[f"foo={queue_length}"],name=f"exp{queue_length}") + results = dvc.experiments.run( + exp_stage.addressing, params=[f"foo={queue_length}"], name=f"exp{queue_length}" + ) ref_info = first(exp_refs_by_rev(scm, first(results))) removed = sorted(dvc.experiments.remove(all_commits=True, queue=True)) @@ -57,6 +62,7 @@ def test_remove_all_experiments_queued_and_commits(tmp_dir, scm, dvc, exp_stage) assert len(dvc.experiments.stash_revs) == 0 assert scm.get_ref(str(ref_info)) is None + def test_remove_special_queued_experiments(tmp_dir, scm, dvc, exp_stage): dvc.experiments.run( exp_stage.addressing, params=["foo=1"], queue=True, name="queue1"