From c911b01246920e0893b7572442ba96deaded8866 Mon Sep 17 00:00:00 2001 From: Teagan Glenn Date: Sat, 31 Jan 2026 08:03:22 -0700 Subject: [PATCH] Fix mock module name --- frigate/test/test_custom_classification.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/frigate/test/test_custom_classification.py b/frigate/test/test_custom_classification.py index 56e1ad766..fd2fda460 100644 --- a/frigate/test/test_custom_classification.py +++ b/frigate/test/test_custom_classification.py @@ -5,9 +5,9 @@ from unittest.mock import MagicMock, patch MOCK_MODULES = [ "frigate.data_processing.real_time.custom_classification.cv2", - "frigate.data_processing.real_time.custom-classification.load_labels", - "frigate.data_processing.real_time.custom-classification.write_classification_attempt", - "frigate.data_processing.real_time.custom-classification.log", + "frigate.data_processing.real_time.custom_classification.load_labels", + "frigate.data_processing.real_time.custom_classification.write_classification_attempt", + "frigate.data_processing.real_time.custom_classification.log", ] AUTO_SPEC_MOCK_MODULES = [ "frigate.data_processing.real_time.custom_classification.Interpreter", @@ -201,6 +201,7 @@ class TestCustomObjectClassificationIntegration(unittest.TestCase): def setUp(self): """Import the processor after mocking dependencies""" + def cvtColor(frame, color): return self.np.zeros((WIDTH, HEIGHT, 3), dtype=self.np.uint8) @@ -219,7 +220,9 @@ class TestCustomObjectClassificationIntegration(unittest.TestCase): self.addCleanup(patcher.stop) patcher.return_value = MagicMock() - mock_cv2 = self.patchers["frigate.data_processing.real_time.custom_classification.cv2"] + mock_cv2 = self.patchers[ + "frigate.data_processing.real_time.custom_classification.cv2" + ] import numpy as np