From 3200953996c529887e3777641ddc812663a9fc68 Mon Sep 17 00:00:00 2001 From: Matt Wong <156021403+mawong-amd@users.noreply.github.com> Date: Tue, 25 Jun 2024 13:53:33 -0500 Subject: [PATCH] Fix linting (#68) --- vllm/distributed/device_communicators/custom_all_reduce.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/vllm/distributed/device_communicators/custom_all_reduce.py b/vllm/distributed/device_communicators/custom_all_reduce.py index a5a60828ada0b..287a8a46bb394 100644 --- a/vllm/distributed/device_communicators/custom_all_reduce.py +++ b/vllm/distributed/device_communicators/custom_all_reduce.py @@ -74,9 +74,8 @@ def _is_full_nvlink(device_ids: List[int], world_size) -> bool: if link_type["hops"] != 1 or link_type["type"] != 2: return False except AmdSmiException as error: - logger.error( - "AMD link detection failed.", - exc_info=error) + logger.error("AMD link detection failed.", + exc_info=error) return False else: handles = [pynvml.nvmlDeviceGetHandleByIndex(i) for i in device_ids]