This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 6b4350e89c Modify `template_fields` of `DatabricksSqlOperator` to
support parent class fields (#32253)
6b4350e89c is described below
commit 6b4350e89cd1b3cc66347b31b10337105ccb9907
Author: eladkal <[email protected]>
AuthorDate: Thu Jun 29 10:36:48 2023 +0300
Modify `template_fields` of `DatabricksSqlOperator` to support parent class
fields (#32253)
---
airflow/providers/databricks/operators/databricks_sql.py | 11 ++++-------
1 file changed, 4 insertions(+), 7 deletions(-)
diff --git a/airflow/providers/databricks/operators/databricks_sql.py
b/airflow/providers/databricks/operators/databricks_sql.py
index 663b4ef0fe..7d9fbb2885 100644
--- a/airflow/providers/databricks/operators/databricks_sql.py
+++ b/airflow/providers/databricks/operators/databricks_sql.py
@@ -70,14 +70,11 @@ class DatabricksSqlOperator(SQLExecuteQueryOperator):
:param csv_params: parameters that will be passed to the
``csv.DictWriter`` class used to write CSV data.
"""
- template_fields: Sequence[str] = (
- "sql",
- "_output_path",
- "schema",
- "catalog",
- "http_headers",
- "databricks_conn_id",
+ template_fields: Sequence[str] = tuple(
+ {"_output_path", "schema", "catalog", "http_headers",
"databricks_conn_id"}
+ | set(SQLExecuteQueryOperator.template_fields)
)
+
template_ext: Sequence[str] = (".sql",)
template_fields_renderers = {"sql": "sql"}
