.

Alexei-V-Ivanov-AMD · May 18, 2024 · f6de88c · f6de88c
1 parent 8715e4e
commit f6de88c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/spec_decode/e2e/conftest.py b/tests/spec_decode/e2e/conftest.py
@@ -9,7 +9,8 @@
 from vllm.utils import is_hip
 
 if (not is_hip()):
-    from pynvml import nvmlDeviceGetHandleByIndex, nvmlDeviceGetMemoryInfo, nvmlInit
+    from pynvml import nvmlDeviceGetHandleByIndex
+    from pynvml import nvmlDeviceGetMemoryInfo, nvmlInit
 
 from vllm import LLM
 from vllm.engine.arg_utils import AsyncEngineArgs