diff options
Diffstat (limited to 'etc/taskcluster/decisionlib.py')
| -rw-r--r-- | etc/taskcluster/decisionlib.py | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/etc/taskcluster/decisionlib.py b/etc/taskcluster/decisionlib.py index 6ba349e8d0d..82d6dcacced 100644 --- a/etc/taskcluster/decisionlib.py +++ b/etc/taskcluster/decisionlib.py @@ -286,6 +286,7 @@ class GenericWorkerTask(Task): super().__init__(*args, **kwargs) self.max_run_time_minutes = 30 self.env = {} + self.features = {} self.mounts = [] self.artifacts = [] @@ -314,6 +315,7 @@ class GenericWorkerTask(Task): worker_payload, env=self.env, mounts=self.mounts, + features=self.features, artifacts=[ { "type": type_, @@ -337,6 +339,15 @@ class GenericWorkerTask(Task): self.artifacts.extend((type, path) for path in paths) return self + def with_features(self, *names): + """ + Enable the given `generic-worker` features. + + <https://github.com/taskcluster/generic-worker/blob/master/native_windows.yml> + """ + self.features.update({name: True for name in names}) + return self + def _mount_content(self, url_or_artifact_name, task_id, sha256): if task_id: content = {"taskId": task_id, "artifact": url_or_artifact_name} @@ -693,9 +704,9 @@ class DockerWorkerTask(UnixTaskMixin, Task): def with_features(self, *names): """ - Enable the give `docker-worker` features. + Enable the given `docker-worker` features. - <https://docs.taskcluster.net/docs/reference/workers/docker-worker/docs/features> + <https://github.com/taskcluster/docker-worker/blob/master/docs/features.md> """ self.features.update({name: True for name in names}) return self |