From 8921255da38d3983fde88b370450fd42fc6433df Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Wed, 28 Feb 2024 15:10:12 -0700 Subject: [PATCH] Fix movement weight test --- frigate/config.py | 6 +++--- frigate/test/test_config.py | 10 +++++----- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/frigate/config.py b/frigate/config.py index 71e12f778..66599f762 100644 --- a/frigate/config.py +++ b/frigate/config.py @@ -189,7 +189,7 @@ class PtzAutotrackConfig(FrigateBaseModel): timeout: int = Field( default=10, title="Seconds to delay before returning to preset." ) - movement_weights: Optional[Union[str, List[float]]] = Field( + movement_weights: Optional[Union[str, List[str]]] = Field( default=[], title="Internal value used for PTZ movements based on the speed of your camera's motor.", ) @@ -204,9 +204,9 @@ class PtzAutotrackConfig(FrigateBaseModel): return None if isinstance(v, str): - weights = list(map(float, v.split(","))) + weights = list(map(str, map(float, v.split(",")))) elif isinstance(v, list): - weights = [float(val) for val in v] + weights = [str(val) for val in v] else: raise ValueError("Invalid type for movement_weights") diff --git a/frigate/test/test_config.py b/frigate/test/test_config.py index b14550cf4..1f25488cc 100644 --- a/frigate/test/test_config.py +++ b/frigate/test/test_config.py @@ -1535,11 +1535,11 @@ class TestConfig(unittest.TestCase): runtime_config = frigate_config.runtime_config() assert runtime_config.cameras["back"].onvif.autotracking.movement_weights == [ - 0.0, - 1.0, - 1.23, - 2.34, - 0.5, + "0.0", + "1.0", + "1.23", + "2.34", + "0.5", ] def test_fails_invalid_movement_weights(self):