administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-05 02:35:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix type errors

Browse Source
This commit is contained in:
Sergey Krashevich 2023-05-03 10:47:49 +03:00
parent dc4b593e32
commit 82eaaaf517
No known key found for this signature in database
GPG Key ID: 625171324E7D3856
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
frigate/stats.py
View File

@ -151,11 +151,12 @@ async def set_gpu_stats(
nvidia_usage = get_nvidia_gpu_stats() nvidia_usage = get_nvidia_gpu_stats()
if nvidia_usage: if nvidia_usage:
for i in nvidia_usage: for i in range(len(nvidia_usage)):
stats[nvidia_usage[i]["name"]] = { stats[nvidia_usage[i]["name"]] = {
"gpu": round(nvidia_usage[i]["gpu"], 2), "gpu": round(float(nvidia_usage[i]["gpu"]), 2),
"mem": round(nvidia_usage[i]["mem"], 2), "mem": round(float(nvidia_usage[i]["mem"]), 2),
} }
else: else:
stats["nvidia-gpu"] = {"gpu": -1, "mem": -1} stats["nvidia-gpu"] = {"gpu": -1, "mem": -1}
hwaccel_errors.append(args) hwaccel_errors.append(args)

2
frigate/util.py
View File

@ -924,7 +924,7 @@ def try_get_info(f, h, default="N/A"):
return v return v
def get_nvidia_gpu_stats() -> dict[str, str]: def get_nvidia_gpu_stats() -> dict[int, dict]:
nvml.nvmlInit() nvml.nvmlInit()
deviceCount = nvml.nvmlDeviceGetCount() deviceCount = nvml.nvmlDeviceGetCount()
results = {} results = {}