Merge pull request #3693 from NicholasTurner23/update/Kafka-implement…

…ations Clean up
airqo-platform · Oct 19, 2024 · 378ee41 · 378ee41
2 parents 3601d23 + 36c4b89
commit 378ee41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/workflows/airqo_etl_utils/message_broker_utils.py b/src/workflows/airqo_etl_utils/message_broker_utils.py
@@ -357,7 +357,7 @@ def publish_to_topic(
                 if not auto_partition:
                     self.partition_loads[selected_partition] += 1
 
-        producer.close()
+        producer.flush()
 
     def consume_from_topic(
         self,