diff --git a/tests/model_executor/models/test_kv_cache_scale_mapper.py b/tests/model_executor/models/test_kv_cache_scale_mapper.py
index bba85e69a..7d7950784 100644
--- a/tests/model_executor/models/test_kv_cache_scale_mapper.py
+++ b/tests/model_executor/models/test_kv_cache_scale_mapper.py
@@ -97,7 +97,7 @@ def test_model_loaders_do_not_call_removed_get_cache_scale(rel_path: str) -> Non
         ),
     ],
 )
-def test_hunyuan_expert_mapping_uses_vllm_public_contract(
+def test_hunyuan_expert_mapping_separates_public_and_local_contracts(
     module_name: str,
     class_name: str,
     monkeypatch: pytest.MonkeyPatch,
diff --git a/vllm_omni/diffusion/models/hunyuan_image3/hunyuan_image3_transformer.py b/vllm_omni/diffusion/models/hunyuan_image3/hunyuan_image3_transformer.py
index dd1d4af0a..d58a43498 100644
--- a/vllm_omni/diffusion/models/hunyuan_image3/hunyuan_image3_transformer.py
+++ b/vllm_omni/diffusion/models/hunyuan_image3/hunyuan_image3_transformer.py
@@ -2205,6 +2205,7 @@ class HunyuanImage3Model(nn.Module):
         )
 
     def _get_expert_weights_remapping(self) -> dict[str, tuple[str, int, int]]:
+        """Return the vLLM-Omni-local remapping for Hunyuan checkpoint weights."""
         if not _is_moe(self.config):
             return {}
         return {
diff --git a/vllm_omni/model_executor/models/hunyuan_image3/hunyuan_image3.py b/vllm_omni/model_executor/models/hunyuan_image3/hunyuan_image3.py
index ead3c6eb3..58b67d8aa 100755
--- a/vllm_omni/model_executor/models/hunyuan_image3/hunyuan_image3.py
+++ b/vllm_omni/model_executor/models/hunyuan_image3/hunyuan_image3.py
@@ -144,6 +144,7 @@ class HunyuanModel(HunYuanModel):
         )
 
     def _get_expert_weights_remapping(self) -> dict[str, tuple[str, int, int]]:
+        """Return the vLLM-Omni-local remapping for Hunyuan checkpoint weights."""
         if not _is_moe(self.config):
             return {}
         return {