Skip to content

Commit ce6ebf6

Browse files
authored
[None][fix] api stability bug in status label (#7861)
Signed-off-by: Yan Chunwei <[email protected]>
1 parent 8fecc06 commit ce6ebf6

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

tensorrt_llm/llmapi/llm_args.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1183,8 +1183,8 @@ class BaseLlmArgs(StrictBaseModel):
11831183
fail_fast_on_attention_window_too_large: bool = Field(
11841184
default=False,
11851185
description=
1186-
"Fail fast when attention window is too large to fit even a single sequence in the KV cache."
1187-
)
1186+
"Fail fast when attention window is too large to fit even a single sequence in the KV cache.",
1187+
status="prototype")
11881188

11891189
# LoRA arguments
11901190
enable_lora: bool = Field(default=False, description="Enable LoRA.")

tests/unittest/api_stability/api_stability_core.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -509,7 +509,9 @@ def check_status(field_name, reference_status, context=""):
509509
# step 1: check the method status
510510
method = getattr(self.TEST_CLASS, method_name)
511511
if method_name in committed_data.get('methods', {}):
512-
continue
512+
if method_name != "__init__":
513+
continue
514+
# Both committed and non-committed methods have __init__ with different parameters
513515
if method_name != "__init__":
514516
method_status = get_api_status(method)
515517
if method_status is None:

0 commit comments

Comments
 (0)