Fix linting (#68)

ROCm · Jun 25, 2024 · 3200953 · 3200953
1 parent 3e7b0b6
commit 3200953
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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]