fix linter

Author: maldwg

src/base/kafka/config.py

@@ -27,12 +27,8 @@
 KAFKA_TRANSACTION_BATCH_CONFIG = CONFIG["environment"].get(
     "kafka_transaction_batch", {}
 )
-KAFKA_TRANSACTION_BATCH_STAGE_CONFIG = KAFKA_TRANSACTION_BATCH_CONFIG.get(
-    "stages", {}
-)
-KAFKA_TRANSACTION_BATCH_TOPIC_CONFIG = KAFKA_TRANSACTION_BATCH_CONFIG.get(
-    "topics", {}
-)
+KAFKA_TRANSACTION_BATCH_STAGE_CONFIG = KAFKA_TRANSACTION_BATCH_CONFIG.get("stages", {})
+KAFKA_TRANSACTION_BATCH_TOPIC_CONFIG = KAFKA_TRANSACTION_BATCH_CONFIG.get("topics", {})
 KAFKA_TRANSACTION_BATCH_SIZE = int(
     os.getenv(
         "KAFKA_TRANSACTION_BATCH_SIZE",
@@ -86,6 +82,7 @@ def transaction_batch_settings(
         timeout_ms = max(0, int(os.environ["KAFKA_TRANSACTION_BATCH_TIMEOUT_MS"]))
     return size, timeout_ms
 
+
 KAFKA_TOPIC_CONFIG = CONFIG["environment"].get("kafka_topics", {})
 KAFKA_TOPIC_DEFAULT_PARTITIONS = int(
     os.getenv("KAFKA_TOPIC_PARTITIONS", KAFKA_TOPIC_CONFIG.get("partitions", 12))

tests/kafka/test_kafka_batch_config.py

@@ -67,9 +67,7 @@ def test_exact_topic_overrides_stage_one_field_at_a_time(self):
                 "pipeline-inspector_to_detector-domainator",
                 stage="data_analysis.detector",
                 stage_config={"detector": {"size": 10, "timeout_ms": 25}},
-                topic_config={
-                    "pipeline-inspector_to_detector-domainator": {"size": 3}
-                },
+                topic_config={"pipeline-inspector_to_detector-domainator": {"size": 3}},
             ),
         )
 

tests/kafka/test_kafka_eos_batch.py

@@ -164,9 +164,7 @@ def test_transaction_can_succeed_after_membership_recovery(
         self.assertEqual(2, producer.begin_transaction.call_count)
 
     @patch("src.base.kafka.producer.Producer")
-    def test_revoked_batch_is_rejected_before_a_transaction_begins(
-        self, mock_producer
-    ):
+    def test_revoked_batch_is_rejected_before_a_transaction_begins(self, mock_producer):
         producer = mock_producer.return_value
         consumer = MagicMock()
         consumer.ensure_batch_assignment_current.side_effect = (

tests/prefilter/test_prefilter.py

@@ -187,15 +187,12 @@ def test_domainator_prefilter_forwards_only_names_with_subdomains(
             [
                 "domain_name",
                 "RegEx",
-                r"^(?=.{1,253}$)((?!-)[A-Za-z0-9-]{1,63}(?<!-)\.)+"
-                r"[A-Za-z]{2,63}$",
+                r"^(?=.{1,253}$)((?!-)[A-Za-z0-9-]{1,63}(?<!-)\.)+" r"[A-Za-z]{2,63}$",
             ],
         ]
         sut = Prefilter(
             consume_topic="pipeline-batch_sender_to_prefilter-domainator_filter",
-            produce_topics=[
-                "pipeline-prefilter_to_inspector-domainator_inspector"
-            ],
+            produce_topics=["pipeline-prefilter_to_inspector-domainator_inspector"],
             relevance_function_name="check_domainator_relevance",
             validation_config=validation_config,
         )