Accept task_key as an argument in DatabricksTaskBaseOperator

apache · Dec 17, 2024 · 0862491 · 0862491
1 parent 1834691
commit 0862491
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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