From 806bff857f9e787a2bae4a0c488785587bd82a17 Mon Sep 17 00:00:00 2001 From: Tai An Date: Wed, 9 Sep 2026 03:11:40 -0700 Subject: [PATCH] [None][fix] correct disagg_utils __all__ entry to extract_disagg_cfg The public API list `__all__` in tensorrt_llm/llmapi/disagg_utils.py references `extract_server_configs`, which is not defined anywhere in the module. The actual extraction entry point is `extract_disagg_cfg`, used internally by `parse_disagg_config_file` and covered by tests/unittest/disaggregated/test_disagg_utils.py. As a result `from tensorrt_llm.llmapi.disagg_utils import *` raises `AttributeError: module ... has no attribute 'extract_server_configs'`. Point the `__all__` entry at the real symbol `extract_disagg_cfg`. Signed-off-by: Tai An --- tensorrt_llm/llmapi/disagg_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorrt_llm/llmapi/disagg_utils.py b/tensorrt_llm/llmapi/disagg_utils.py index d1851364d7c5..94510bd8f62f 100644 --- a/tensorrt_llm/llmapi/disagg_utils.py +++ b/tensorrt_llm/llmapi/disagg_utils.py @@ -15,7 +15,7 @@ __all__ = [ 'ServerConfig', 'parse_disagg_config_file', - 'extract_server_configs', + 'extract_disagg_cfg', 'split_world_comm', 'get_usage_tokens_from_ctx', 'rewrite_usage_info_from_ctx',