From 24a96d0e0838c7ef28fb8191993bbea4cb981dec Mon Sep 17 00:00:00 2001
From: Teagan Glenn <that@teagantotally.rocks>
Date: Sat, 31 Jan 2026 05:43:51 -0700
Subject: [PATCH] Fix formatting

---
 frigate/test/test_custom_classification.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/frigate/test/test_custom_classification.py b/frigate/test/test_custom_classification.py
index 3a0644c48..a554c5dba 100644
--- a/frigate/test/test_custom_classification.py
+++ b/frigate/test/test_custom_classification.py
@@ -26,11 +26,8 @@ MOCK_MODULES = [
     "tensorflow.lite.python.interpreter",
 ]
 
-ORIGINAL_MODULES = {
-    mod: sys.modules[mod]
-    for mod
-    in MOCK_MODULES
-}
+ORIGINAL_MODULES = {mod: sys.modules[mod] for mod in MOCK_MODULES}}
+
 
 class TestCustomObjectClassificationZones(unittest.TestCase):
     """Test that zone information is correctly added to custom classification MQTT messages"""