(airflow) branch main updated: Accept BLOCKED and WAITING_FOR_RETRY Databricks run states (#69193)

[amoghdesai]

This is an automated email from the ASF dual-hosted git repository.

amoghrajesh pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new a3f51fcd9ce Accept BLOCKED and WAITING_FOR_RETRY Databricks run states 
(#69193)
a3f51fcd9ce is described below

commit a3f51fcd9ce8b90186578d99a16c9b465bb88cb6
Author: Amogh Desai <amoghrajesh1...@gmail.com>
AuthorDate: Wed Jul 1 12:46:17 2026 +0530

    Accept BLOCKED and WAITING_FOR_RETRY Databricks run states (#69193)
---
 .../databricks/src/airflow/providers/databricks/hooks/databricks.py     | 2 ++
 providers/databricks/tests/unit/databricks/hooks/test_databricks.py     | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git 
a/providers/databricks/src/airflow/providers/databricks/hooks/databricks.py 
b/providers/databricks/src/airflow/providers/databricks/hooks/databricks.py
index 7bba7634552..3affebbba91 100644
--- a/providers/databricks/src/airflow/providers/databricks/hooks/databricks.py
+++ b/providers/databricks/src/airflow/providers/databricks/hooks/databricks.py
@@ -99,6 +99,8 @@ class RunState:
         "TERMINATED",
         "SKIPPED",
         "INTERNAL_ERROR",
+        "BLOCKED",
+        "WAITING_FOR_RETRY",
         "QUEUED",
     ]
 
diff --git 
a/providers/databricks/tests/unit/databricks/hooks/test_databricks.py 
b/providers/databricks/tests/unit/databricks/hooks/test_databricks.py
index 2b79d9b1044..9086e291986 100644
--- a/providers/databricks/tests/unit/databricks/hooks/test_databricks.py
+++ b/providers/databricks/tests/unit/databricks/hooks/test_databricks.py
@@ -1560,7 +1560,7 @@ class TestRunState:
             assert run_state.is_terminal
 
     def test_is_terminal_false(self):
-        non_terminal_states = ["PENDING", "RUNNING", "TERMINATING", "QUEUED"]
+        non_terminal_states = ["PENDING", "RUNNING", "TERMINATING", "QUEUED", 
"BLOCKED", "WAITING_FOR_RETRY"]
         for state in non_terminal_states:
             run_state = RunState(state, "", "")
             assert not run_state.is_terminal