[CI] Auto-label pooling PRs and issues (#59913)

Signed-off-by: Taneem Ibrahim <taneem.ibrahim@gmail.com>
Co-authored-by: Lucas Wilkinson <LucasWilkinson@users.noreply.github.com>
This commit is contained in:
Taneem Ibrahim
2026-10-04 13:26:10 +00:00
committed by GitHub
co-authored by Lucas Wilkinson
parent d61081dc3d
commit d64f6cd08a
2 changed files with 47 additions and 0 deletions
+26
View File
@@ -112,6 +112,32 @@ pull_request_rules:
add:
- frontend
- name: label-pooling
description: Automatically apply pooling label
conditions:
- -closed
- or:
- files~=^vllm/entrypoints/pooling/
- files~=^vllm/model_executor/layers/pooler/
- files=vllm/model_executor/models/transformers/pooling.py
- files=vllm/model_executor/models/adapters.py
- files=vllm/config/pooler.py
- files=vllm/pooling_params.py
- files~=^vllm/v1/pool/
- files~=^vllm/v1/worker/gpu/pool/
- files~=^tests/entrypoints/pooling/
- files~=^tests/models/(language|multimodal)/pooling(_mteb_test)?/
- files~=^tests/model_executor/layers/test_pooler_.*\.py$
- files=tests/test_pooling_params.py
- files~=^tests/v1/e2e/general/test_pooling_.*\.py$
- files~=^docs/models/pooling_models/
- files~=^examples/pooling/
- title~=(?i)\bpooling\b
actions:
label:
add:
- pooling
- name: label-rust
description: Automatically apply rust label
conditions:
+21
View File
@@ -255,6 +255,27 @@ jobs:
// phrases go in `substrings`: they are specific enough on their
// own, and a boundary match would miss ordinary variations --
// "structured outputs", "tool parsers", "prefix caching".
pooling: {
keywords: [
{ term: "pooling", searchIn: "title" },
{ term: "pooler", searchIn: "title" },
{ term: "rerank", searchIn: "title" },
{ term: "reranker", searchIn: "title" },
{ term: "reranking", searchIn: "title" },
],
substrings: [
{ term: "embedding model", searchIn: "title" },
{ term: "embedding API", searchIn: "title" },
{ term: "embeddings API", searchIn: "title" },
{ term: "sentence embedding", searchIn: "title" },
{ term: "sequence classification", searchIn: "title" },
{ term: "token classification", searchIn: "title" },
{ term: "cross-encoder", searchIn: "title" },
{ term: "late interaction", searchIn: "title" },
{ term: "late-interaction", searchIn: "title" },
{ term: "reward model", searchIn: "title" },
],
},
"speculative-decoding": {
keywords: [
{ term: "eagle", searchIn: "title" },