[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:
co-authored by
Lucas Wilkinson
parent
d61081dc3d
commit
d64f6cd08a
@@ -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:
|
||||
|
||||
@@ -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" },
|
||||
|
||||
Reference in New Issue
Block a user