lint

Natalia Gimelshein · Natalia Gimelshein · commit 9c069ee157ff · 2020-09-13T16:35:17.000-07:00
diff --git a/torch/cuda/memory.py b/torch/cuda/memory.py
@@ -483,7 +483,7 @@ def list_gpu_processes(device: Union[Device, int] = None) -> str:
     """
 
     try:
-        import pynvml # type: ignore
+        import pynvml  # type: ignore
     except ModuleNotFoundError:
         return("pynvml module not found, please install pynvml")
     from pynvml import NVMLError_DriverNotLoaded
