Expand exception handling for NVML errors (#22)

Extended the exception handling to catch any general exception along with `NVML_ERROR_NOT_SUPPORTED`.
Lenochxd · Jul 23, 2024 · f93ca52 · f93ca52
1 parent 2fd3595
commit f93ca52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app/buttons/usage/get_usage.py b/app/buttons/usage/get_usage.py
@@ -80,7 +80,7 @@ def get_usage(
                     computer_info["gpus"][f"GPU{count + 1}"] = {
                         "usage_percent": int(utilization.gpu),
                     }
-            except pynvml.nvml.NVML_ERROR_NOT_SUPPORTED:
+            except (pynvml.nvml.NVML_ERROR_NOT_SUPPORTED, Exception):
                 computer_info["gpus"]["defaultGPU"] = {}
 
                 config["settings"]["gpu_method"] = "None"