diff --git a/providers/databricks/src/airflow/providers/databricks/plugins/databricks_workflow.py b/providers/databricks/src/airflow/providers/databricks/plugins/databricks_workflow.py index 1b6466f7678b7..16ea7a0b6113e 100644 --- a/providers/databricks/src/airflow/providers/databricks/plugins/databricks_workflow.py +++ b/providers/databricks/src/airflow/providers/databricks/plugins/databricks_workflow.py @@ -154,9 +154,11 @@ def _repair_task( "run_id": databricks_run_id, "latest_repair_id": repair_history_id, "rerun_tasks": tasks_to_repair, - **run_data.get("overriding_parameters", {}), } + if "overriding_parameters" in run_data: + repair_json["overriding_parameters"] = run_data["overriding_parameters"] + return hook.repair_run(repair_json) diff --git a/providers/databricks/tests/unit/databricks/plugins/test_databricks_workflow.py b/providers/databricks/tests/unit/databricks/plugins/test_databricks_workflow.py index 638c3afe1cb26..3c6f57c1d9d5d 100644 --- a/providers/databricks/tests/unit/databricks/plugins/test_databricks_workflow.py +++ b/providers/databricks/tests/unit/databricks/plugins/test_databricks_workflow.py @@ -114,6 +114,7 @@ def test_repair_task_with_params(mock_databricks_hook): expected_payload = { "run_id": DATABRICKS_RUN_ID, "rerun_tasks": tasks_to_repair, + "latest_repair_id": 100, "overriding_parameters": { "key1": "value1", "key2": "value2",