Fix formatting and type checking issues

beeldengeluid · Nov 7, 2024 · c91e061 · c91e061
1 parent a08094d
commit c91e061
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/gpu_measure.py b/gpu_measure.py
@@ -30,7 +30,9 @@ def __init__(self):
     def start_measure_gpu_mem(self):
         def _get_mem_usage():
             while True:
-                self.gpu_mem_usage.append(nvml.nvmlDeviceGetMemoryInfo(self.handle).used >> 20)
+                self.gpu_mem_usage.append(
+                    nvml.nvmlDeviceGetMemoryInfo(self.handle).used >> 20
+                )
                 time.sleep(0.5)
 
                 if self.stop:

diff --git a/pyproject.toml b/pyproject.toml
@@ -76,6 +76,7 @@ exclude = "misc"
 module = [
   'yaml',
   'faster_whisper.*',
-  'boto3.*'
+  'boto3.*',
+  'py3nvml.*'
 ]
 ignore_missing_imports = true