From fa59fb13f4353f0c3f4fd585f17d891ef1eca545 Mon Sep 17 00:00:00 2001 From: Rob Arnold <326113+robarnold@users.noreply.github.com> Date: Sat, 17 Feb 2024 05:43:20 -0800 Subject: [PATCH] Fix formatting --- frigate/util/services.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/frigate/util/services.py b/frigate/util/services.py index 1fcb31077..59cc4d27d 100644 --- a/frigate/util/services.py +++ b/frigate/util/services.py @@ -15,7 +15,12 @@ import psutil import py3nvml.py3nvml as nvml import requests -from frigate.const import DRIVER_AMD, DRIVER_ENV_VAR, FFMPEG_HWACCEL_NVIDIA, FFMPEG_HWACCEL_VAAPI +from frigate.const import ( + DRIVER_AMD, + DRIVER_ENV_VAR, + FFMPEG_HWACCEL_NVIDIA, + FFMPEG_HWACCEL_VAAPI, +) from frigate.util.builtin import clean_camera_user_pass, escape_special_characters logger = logging.getLogger(__name__) @@ -195,6 +200,7 @@ def get_bandwidth_stats(config) -> dict[str, dict]: return usages + def is_vaapi_amd_driver() -> bool: driver = os.environ.get(DRIVER_ENV_VAR) if driver: @@ -207,6 +213,7 @@ def is_vaapi_amd_driver() -> bool: return False else: output = p.stdout.decode('unicode_escape').split("\n") + # VA Info will print out the friendly name of the driver return any("AMD Radeon Graphics" in line for line in output)