--- /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/_features.py	2025-10-02 20:53:24.201799+00:00
+++ /home/runner/work/TensorRT/TensorRT/py/torch_tensorrt/_features.py	2025-10-02 20:53:59.151475+00:00
@@ -165,10 +165,11 @@
def needs_trtllm_for_nccl(f: Callable[..., Any]) -> Callable[..., Any]:
    def wrapper(*args: List[Any], **kwargs: Dict[str, Any]) -> Any:
        if ENABLED_FEATURES.trtllm_for_nccl:
            return f(*args, **kwargs)
        else:
+
            def not_implemented(*args: List[Any], **kwargs: Dict[str, Any]) -> Any:
                raise NotImplementedError(
                    "Refit feature is currently not available in Python 3.13 or higher"
                )