chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

Allow for ".yaml" (#2244)

Browse Source 
* allow for ".yaml"

* remove unused import
This commit is contained in:
Justin Goette
2021-12-12 10:27:05 -05:00
committed by Blake Blackshear
parent 8b2622a234
commit f91f4f0053
3 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/config.py
View File

@@ -12,7 +12,7 @@ import yaml
from pydantic import BaseModel, Extra, Field, validator
from pydantic.fields import PrivateAttr
from frigate.const import BASE_DIR, CACHE_DIR
from frigate.const import BASE_DIR, CACHE_DIR, YAML_EXT
from frigate.edgetpu import load_labels
from frigate.util import create_mask, deep_merge
@@ -864,7 +864,7 @@ class FrigateConfig(FrigateBaseModel):
with open(config_file) as f:
raw_config = f.read()
if config_file.endswith(".yml"):
if config_file.endswith(YAML_EXT):
config = yaml.safe_load(raw_config)
elif config_file.endswith(".json"):
config = json.loads(raw_config)