diff --git a/dvc/testing/benchmarks/cli/commands/test_update.py b/dvc/testing/benchmarks/cli/commands/test_update.py index 86c96679a..3ca4b8aa2 100644 --- a/dvc/testing/benchmarks/cli/commands/test_update.py +++ b/dvc/testing/benchmarks/cli/commands/test_update.py @@ -3,7 +3,7 @@ def test_update(bench_dvc, tmp_dir, dvc, make_dataset): dataset = make_dataset( cache=False, files=True, dvcfile=False, commit=False, remote=False ) - dvc.imp_url(str(dataset), "new") + bench_dvc("import-url", str(dataset), "new") (dataset / "new").write_text("new") bench_dvc("update", "new.dvc") bench_dvc("update", "new.dvc", name="noop") diff --git a/dvc/testing/benchmarks/cli/stories/test_modify_data.py b/dvc/testing/benchmarks/cli/stories/test_modify_data.py index cbb0d11cf..53fe96ae0 100644 --- a/dvc/testing/benchmarks/cli/stories/test_modify_data.py +++ b/dvc/testing/benchmarks/cli/stories/test_modify_data.py @@ -22,8 +22,8 @@ def test_partial_add(bench_dvc, tmp_dir, dvc, dataset, remote): shutil.move(dataset / f, tmp_dir / "partial-copy" / f) # Add/push partial dataset - dvc.add(str(dataset)) - dvc.push() + bench_dvc("add", dataset) + bench_dvc("push") # Add more files to the dataset shutil.copytree("partial-copy", dataset, dirs_exist_ok=True) @@ -39,8 +39,8 @@ def test_partial_add(bench_dvc, tmp_dir, dvc, dataset, remote): def test_partial_remove(bench_dvc, tmp_dir, dvc, dataset, remote): random.seed(5232) # Add/push full dataset - dvc.add(str(dataset)) - dvc.push() + bench_dvc("add", dataset) + bench_dvc("push") # Remove some files for f in glob.glob("*", root_dir=dataset, recursive=True): # type: ignore[call-arg]