From f5abe506fc6ba22a8b8efb0d4784dd85eaa90af4 Mon Sep 17 00:00:00 2001 From: Wei Lee Date: Tue, 19 Nov 2024 09:55:26 +0800 Subject: [PATCH] fix(providers/databricks): remove additional argument passed to repair_run (#44140) --- .../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 03da528468d90..60800ff22e3fe 100644 --- a/providers/src/airflow/providers/databricks/operators/databricks.py +++ b/providers/src/airflow/providers/databricks/operators/databricks.py @@ -121,7 +121,7 @@ def _handle_databricks_operator_execution(operator, hook, log, context) -> None: repair_json = {"run_id": operator.run_id, "rerun_all_failed_tasks": True} if latest_repair_id is not None: repair_json["latest_repair_id"] = latest_repair_id - operator.json["latest_repair_id"] = hook.repair_run(operator, repair_json) + operator.json["latest_repair_id"] = hook.repair_run(repair_json) _handle_databricks_operator_execution(operator, hook, log, context) raise AirflowException(error_message)