Add template fields to ProduceToTopicOperator (#6)

@sergeyh
astronomer · Aug 3, 2022 · 287683e · 287683e
1 parent de3afa2
commit 287683e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/airflow_provider_kafka/operators/produce_to_topic.py b/airflow_provider_kafka/operators/produce_to_topic.py
@@ -46,6 +46,8 @@ class ProduceToTopicOperator(BaseOperator):
     :type poll_timeout: float, optional
     :raises AirflowException: _description_
     """
+
+    template_fields = ('topic', 'producer_function', 'producer_function_args', 'producer_function_kwargs')
 
     def __init__(
         self,