TensorRT-LLMs/tests/unittest/api_stability/references/logits_processor.yaml
Enwei Zhu 224469b096
test: [TRTLLM-4334] Create 1.0 criteria scope from API stability references (#3069)
* committed APIs validation

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* fix

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* clean name

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* separate

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* add TODOs

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* fix naming

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

* fix

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>

---------

Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>
2025-03-26 18:14:35 +08:00

21 lines
471 B
YAML

methods:
__call__:
parameters:
client_id:
annotation: Optional[int]
default: inspect._empty
logits:
annotation: torch.Tensor
default: inspect._empty
req_id:
annotation: int
default: inspect._empty
stream_ptr:
annotation: int
default: inspect._empty
token_ids:
annotation: List[List[int]]
default: inspect._empty
return_annotation: None
properties: {}