update: support --jobs even without --to-remote (#5434)

This commit is contained in:
Batuhan Taskaya
2021-02-09 17:58:59 +02:00
committed by GitHub
parent b370465209
commit 384a4113ca
3 changed files with 19 additions and 3 deletions
+7
View File
@@ -1,3 +1,5 @@
from dvc.exceptions import InvalidArgumentError
from . import locked
@@ -19,6 +21,11 @@ def update(
if isinstance(targets, str):
targets = [targets]
if not to_remote and remote:
raise InvalidArgumentError(
"--remote can't be used without --to-remote"
)
stages = set()
for target in targets:
stages.update(self.stage.collect(target, recursive=recursive))
+1 -1
View File
@@ -18,7 +18,7 @@ def update_import(stage, rev=None, to_remote=False, remote=None, jobs=None):
if to_remote:
_update_import_on_remote(stage, remote, jobs)
else:
stage.reproduce()
stage.reproduce(jobs=jobs)
finally:
stage.frozen = frozen
+11 -2
View File
@@ -4,7 +4,16 @@ from dvc.command.update import CmdUpdate
def test_update(dvc, mocker):
cli_args = parse_args(
["update", "target1", "target2", "--rev", "REV", "--recursive"]
[
"update",
"target1",
"target2",
"--rev",
"REV",
"--recursive",
"-j",
"8",
]
)
assert cli_args.func == CmdUpdate
cmd = cli_args.func(cli_args)
@@ -18,7 +27,7 @@ def test_update(dvc, mocker):
recursive=True,
to_remote=False,
remote=None,
jobs=None,
jobs=8,
)