Update WeightTransferConfig to be more standard like the others (#33989)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>

vllm/config/weight_transfer.py

@@ -1,13 +1,11 @@
 # SPDX-License-Identifier: Apache-2.0
 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
-from dataclasses import dataclass
 from typing import Literal
 
 from vllm.config.utils import config
 
 
 @config
-@dataclass
 class WeightTransferConfig:
     """Configuration for weight transfer during RL training."""
 

vllm/engine/arg_utils.py

@@ -582,10 +582,10 @@ class EngineArgs:
     kv_offloading_backend: KVOffloadingBackend = CacheConfig.kv_offloading_backend
     tokens_only: bool = False
 
-    weight_transfer_config: WeightTransferConfig | None = None
-    """Configuration for weight transfer during RL training. 
-    Accepts a JSON string or dict with backend-specific options.
-    Example: '{"backend": "nccl"}'"""
+    weight_transfer_config: WeightTransferConfig | None = get_field(
+        VllmConfig,
+        "weight_transfer_config",
+    )
 
     def __post_init__(self):
         # support `EngineArgs(compilation_config={...})`