diff --git a/tensorrt_llm/_torch/disaggregation/native/rank_info.py b/tensorrt_llm/_torch/disaggregation/native/rank_info.py index 007e819b8f..bc9ce350c0 100644 --- a/tensorrt_llm/_torch/disaggregation/native/rank_info.py +++ b/tensorrt_llm/_torch/disaggregation/native/rank_info.py @@ -3,7 +3,7 @@ from typing import List, Optional import msgpack -from tensorrt_llm._torch.disaggregation.native.region.aux import AuxBufferMeta +from tensorrt_llm._torch.disaggregation.native.region.aux_ import AuxBufferMeta @dataclass diff --git a/tensorrt_llm/_torch/disaggregation/native/region/aux.py b/tensorrt_llm/_torch/disaggregation/native/region/aux_.py similarity index 100% rename from tensorrt_llm/_torch/disaggregation/native/region/aux.py rename to tensorrt_llm/_torch/disaggregation/native/region/aux_.py