mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-04 07:18:27 +01:00
update: support --jobs even without --to-remote (#5434)
This commit is contained in:
@@ -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))
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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,
|
||||
)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user