From 205f4a1a23f25f51d955d341880efaa85a4fbe03 Mon Sep 17 00:00:00 2001
From: Teagan Glenn <that@teagantotally.rocks>
Date: Sat, 31 Jan 2026 08:09:01 -0700
Subject: [PATCH] Mock suppress_stderr_during

---
 frigate/test/test_custom_classification.py | 9 +--------
 1 file changed, 1 insertion(+), 8 deletions(-)

diff --git a/frigate/test/test_custom_classification.py b/frigate/test/test_custom_classification.py
index fd2fda460..1e99132b2 100644
--- a/frigate/test/test_custom_classification.py
+++ b/frigate/test/test_custom_classification.py
@@ -7,7 +7,7 @@ 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.suppress_stderr_during",
 ]
 AUTO_SPEC_MOCK_MODULES = [
     "frigate.data_processing.real_time.custom_classification.Interpreter",
@@ -240,13 +240,6 @@ class TestCustomObjectClassificationIntegration(unittest.TestCase):
             # If imports fail, skip these tests (they need full Docker environment)
             self.skipTest(f"Requires full Frigate environment: {e}")
 
-    def tearDown(self):
-        for mod in MOCK_MODULES:
-            if mod in ORIGINAL_MODULES:
-                sys.modules[mod] = mod
-            else:
-                del sys.modules[mod]
-
     def test_process_frame_with_zones_includes_zones_in_mqtt(self):
         """
         Integration test: Actually call process_frame() and verify zones in MQTT.