diff --git a/taskcluster/fenix_taskgraph/transforms/version_bump.py b/taskcluster/fenix_taskgraph/transforms/version_bump.py index 9e380e2fe1..dc77cb9442 100644 --- a/taskcluster/fenix_taskgraph/transforms/version_bump.py +++ b/taskcluster/fenix_taskgraph/transforms/version_bump.py @@ -34,7 +34,9 @@ def resolve_keys(config, tasks): @transforms.add def build_worker_definition(config, tasks): for task in tasks: - worker_definition = {} + worker_definition = { + "branch": config.params["head_ref"].decode("utf-8"), + } task["worker"].update(worker_definition) yield task diff --git a/taskcluster/fenix_taskgraph/worker_types.py b/taskcluster/fenix_taskgraph/worker_types.py index 9200377286..9017932106 100644 --- a/taskcluster/fenix_taskgraph/worker_types.py +++ b/taskcluster/fenix_taskgraph/worker_types.py @@ -172,6 +172,7 @@ def build_github_release_payload(config, task, task_def): Required("bump"): bool, Optional("bump-files"): [text_type], Optional("push"): bool, + Optional("branch"): text_type, }, ) def build_version_bump_payload(config, task, task_def): @@ -196,3 +197,6 @@ def build_version_bump_payload(config, task, task_def): if worker.get('force-dry-run'): task_def['payload']['dry_run'] = True + + if worker.get("branch"): + task_def["payload"]["branch"] = worker["branch"]