diff options
Diffstat (limited to 'tools/tryselect/selectors/fuzzy.py')
| -rw-r--r-- | tools/tryselect/selectors/fuzzy.py | 7 |
1 files changed, 6 insertions, 1 deletions
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: |