Update await_message.py (#8)

Adding templated fields to operator
astronomer · Aug 3, 2022 · f068654 · f068654
1 parent 7914f6a
commit f068654
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/airflow_provider_kafka/operators/await_message.py b/airflow_provider_kafka/operators/await_message.py
@@ -45,6 +45,8 @@ class AwaitKafkaMessageOperator(BaseOperator):
     BLUE = "#ffefeb"
     ui_color = BLUE
 
+    template_fields = ('topics', 'apply_function', 'apply_function_args', 'apply_function_kwargs')
+
     def __init__(
         self,
         topics: Sequence[str],