.

Author: Alexei-V-Ivanov-AMD

tests/spec_decode/e2e/conftest.py

@@ -9,8 +9,8 @@
 from vllm.utils import is_hip
 
 if (not is_hip()):
-    from pynvml import nvmlDeviceGetHandleByIndex
-    from pynvml import nvmlDeviceGetMemoryInfo, nvmlInit
+    from pynvml import (nvmlDeviceGetHandleByIndex
+                    nvmlDeviceGetMemoryInfo, nvmlInit)
 
 from vllm import LLM
 from vllm.engine.arg_utils import AsyncEngineArgs