Fix formatting

This commit is contained in:
Rob Arnold 2024-02-17 05:43:20 -08:00
parent f9b17d3f00
commit fa59fb13f4

View File

@ -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)