From 08624911481cef20911bcf7330092bce1594fcc6 Mon Sep 17 00:00:00 2001
From: Idris Adebisi <idris.adebisi@yahoo.com>
Date: Tue, 17 Dec 2024 13:10:16 +0000
Subject: [PATCH] Accept task_key as an argument in DatabricksTaskBaseOperator

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

diff --git a/providers/src/airflow/providers/databricks/operators/databricks.py b/providers/src/airflow/providers/databricks/operators/databricks.py
index 83435ee9db13e..c9bf6ba859fc0 100644
--- a/providers/src/airflow/providers/databricks/operators/databricks.py
+++ b/providers/src/airflow/providers/databricks/operators/databricks.py
@@ -1053,7 +1053,7 @@ def _generate_databricks_task_key(self, task_id: str | None = None) -> str:
             task_id = task_id or self.task_id
             task_key = f"{self.dag_id}__{task_id}".encode()
             self._databricks_task_key = hashlib.md5(task_key).hexdigest()
-            self.log.info("Generated databricks task key: %s", task_key)
+            self.log.info("Generated databricks task key: %s", self._databricks_task_key)
         return self._databricks_task_key
 
     @property