From 85e812ff4bd35fa007612b4f43f06c681840f4f1 Mon Sep 17 00:00:00 2001
From: Jake Roach <116606359+jroachgolf84@users.noreply.github.com>
Date: Sat, 25 Apr 2026 21:46:59 -0400
Subject: [PATCH 1/2] issue-46529: Adding task_config to template_fields for
 DatabricksTaskOperator

---
 .../src/airflow/providers/databricks/operators/databricks.py    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/providers/databricks/src/airflow/providers/databricks/operators/databricks.py b/providers/databricks/src/airflow/providers/databricks/operators/databricks.py
index 826d79c4bdefd..13076c79bda0e 100644
--- a/providers/databricks/src/airflow/providers/databricks/operators/databricks.py
+++ b/providers/databricks/src/airflow/providers/databricks/operators/databricks.py
@@ -1666,7 +1666,7 @@ class DatabricksTaskOperator(DatabricksTaskBaseOperator):
     """
 
     CALLER = "DatabricksTaskOperator"
-    template_fields = ("workflow_run_metadata",)
+    template_fields = ("task_config", "workflow_run_metadata",)
 
     def __init__(
         self,

From a3afac9b71da51221e60bcd76452d65f2cd1837b Mon Sep 17 00:00:00 2001
From: Jake Roach <116606359+jroachgolf84@users.noreply.github.com>
Date: Sun, 26 Apr 2026 07:44:08 -0400
Subject: [PATCH 2/2] issue-46529: Fixing static checks

---
 .../airflow/providers/databricks/operators/databricks.py    | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/providers/databricks/src/airflow/providers/databricks/operators/databricks.py b/providers/databricks/src/airflow/providers/databricks/operators/databricks.py
index 13076c79bda0e..788a12f1a3641 100644
--- a/providers/databricks/src/airflow/providers/databricks/operators/databricks.py
+++ b/providers/databricks/src/airflow/providers/databricks/operators/databricks.py
@@ -1666,7 +1666,11 @@ class DatabricksTaskOperator(DatabricksTaskBaseOperator):
     """
 
     CALLER = "DatabricksTaskOperator"
-    template_fields = ("task_config", "workflow_run_metadata",)
+    template_fields = (
+        "databricks_conn_id",
+        "task_config",
+        "workflow_run_metadata",
+    )
 
     def __init__(
         self,