12
edits
m (→Testing: fixing links) |
No edit summary |
||
Line 272: | Line 272: | ||
--- a/taskcluster/taskgraph/transforms/task.py | --- a/taskcluster/taskgraph/transforms/task.py | ||
+++ b/taskcluster/taskgraph/transforms/task.py | +++ b/taskcluster/taskgraph/transforms/task.py | ||
@@ - | @@ -1717,16 +1717,48 @@ def add_index_routes(config, tasks): | ||
index_type = index.get('type', 'generic') | index_type = index.get('type', 'generic') | ||
task = index_builders[index_type](config, task) | task = index_builders[index_type](config, task) | ||
del task['index'] | del task['index'] | ||
yield task | yield task | ||
@transforms.add | @transforms.add | ||
+def | +def set_worker_type(config, tasks): | ||
+ for task in tasks: | + for task in tasks: | ||
+ level = str(config.params['level']) | + level = str(config.params['level']) | ||
+ task['worker-type'] | + | ||
+ provisioner_id, worker_type = get_worker_type( | |||
+ config.graph_config, | |||
+ task['worker-type'], | |||
+ level, | |||
+ ) | |||
+ task['worker-type'] = '/'.join([provisioner_id, worker_type]) | |||
+ yield task | + yield task | ||
+ | |||
+ | + | ||
+@transforms.add | +@transforms.add | ||
Line 300: | Line 304: | ||
+ find_replace_dict = { | + find_replace_dict = { | ||
+ "releng-hardware/gecko-t-osx-1010": "releng-hardware/gecko-t-osx-1010-beta", | + "releng-hardware/gecko-t-osx-1010": "releng-hardware/gecko-t-osx-1010-beta", | ||
+ "releng-hardware/gecko-t-linux-talos": "releng-hardware/gecko-t-linux-talos-b" | |||
+ } | + } | ||
+ for task in tasks: | + for task in tasks: | ||
+ if task['worker-type'] in find_replace_dict: | + if task['worker-type'] in find_replace_dict: | ||
+ task['worker-type'] = find_replace_dict[task['worker-type']] | + task['worker-type'] = find_replace_dict[task['worker-type']] | ||
+ yield task | + yield task | ||
+ | |||
+ | + | ||
+@transforms.add | +@transforms.add | ||
Line 312: | Line 316: | ||
for task in tasks: | for task in tasks: | ||
level = str(config.params['level']) | level = str(config.params['level']) | ||
provisioner_id, worker_type = | provisioner_id, worker_type = get_worker_type( | ||
config.graph_config, | |||
task['worker-type'], | |||
level, | |||
</pre> | </pre> | ||
edits