From 82eaaaf51780bf37bf50503f193a40293fa67abd Mon Sep 17 00:00:00 2001 From: Sergey Krashevich Date: Wed, 3 May 2023 10:47:49 +0300 Subject: [PATCH] fix type errors --- frigate/stats.py | 7 ++++--- frigate/util.py | 2 +- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/frigate/stats.py b/frigate/stats.py index 32a96cd04..d3ffc9204 100644 --- a/frigate/stats.py +++ b/frigate/stats.py @@ -151,11 +151,12 @@ async def set_gpu_stats( nvidia_usage = get_nvidia_gpu_stats() if nvidia_usage: - for i in nvidia_usage: + for i in range(len(nvidia_usage)): stats[nvidia_usage[i]["name"]] = { - "gpu": round(nvidia_usage[i]["gpu"], 2), - "mem": round(nvidia_usage[i]["mem"], 2), + "gpu": round(float(nvidia_usage[i]["gpu"]), 2), + "mem": round(float(nvidia_usage[i]["mem"]), 2), } + else: stats["nvidia-gpu"] = {"gpu": -1, "mem": -1} hwaccel_errors.append(args) diff --git a/frigate/util.py b/frigate/util.py index 60584df77..f805fbdca 100755 --- a/frigate/util.py +++ b/frigate/util.py @@ -924,7 +924,7 @@ def try_get_info(f, h, default="N/A"): return v -def get_nvidia_gpu_stats() -> dict[str, str]: +def get_nvidia_gpu_stats() -> dict[int, dict]: nvml.nvmlInit() deviceCount = nvml.nvmlDeviceGetCount() results = {}