From fbaf0bb26397aa498eb9156f06d5a6fe34dd7dd8 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Fri, 19 Apr 2024 03:14:29 +0200 Subject: Merging upstream version 125.0.1. Signed-off-by: Daniel Baumann --- tools/tryselect/selectors/fuzzy.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'tools/tryselect/selectors/fuzzy.py') diff --git a/tools/tryselect/selectors/fuzzy.py b/tools/tryselect/selectors/fuzzy.py index 7a9bccc4b7..a9b139fa6f 100644 --- a/tools/tryselect/selectors/fuzzy.py +++ b/tools/tryselect/selectors/fuzzy.py @@ -12,7 +12,7 @@ from mach.util import get_state_dir from ..cli import BaseTryParser from ..push import check_working_directory, generate_try_task_config, push_to_try -from ..tasks import filter_tasks_by_paths, generate_tasks +from ..tasks import filter_tasks_by_paths, filter_tasks_by_worker_type, generate_tasks from ..util.fzf import ( FZF_NOT_FOUND, PREVIEW_SCRIPT, @@ -182,6 +182,11 @@ def run( if filter_by_uncommon_try_tasks(task_name) } + if try_config_params.get("try_task_config", {}).get("worker-types", []): + all_tasks = filter_tasks_by_worker_type(all_tasks, try_config_params) + if not all_tasks: + return 1 + if test_paths: all_tasks = filter_tasks_by_paths(all_tasks, test_paths) if not all_tasks: -- cgit v1.2.3