[GitHub] [airflow] pankajastro commented on a diff in pull request #30744: Add retry param in databrics async operator

["pankajastro (via GitHub)" <gi...@apache.org>]

pankajastro commented on code in PR #30744:
URL: https://github.com/apache/airflow/pull/30744#discussion_r1174622439


##########
airflow/providers/databricks/triggers/databricks.py:
##########
@@ -32,14 +32,30 @@ class DatabricksExecutionTrigger(BaseTrigger):
     :param databricks_conn_id: Reference to the :ref:`Databricks connection <howto/connection:databricks>`.
     :param polling_period_seconds: Controls the rate of the poll for the result of this run.
         By default, the trigger will poll every 30 seconds.
+    :param retry_limit: The number of times to retry the connection in case of service outages.
+    :param retry_delay: The number of seconds to wait between retries.
+    :param run_page_url: The run page url.
     """
 
-    def __init__(self, run_id: int, databricks_conn_id: str, polling_period_seconds: int = 30) -> None:
+    def __init__(
+        self,
+        run_id: int,
+        databricks_conn_id: str,
+        polling_period_seconds: int = 30,
+        retry_limit: int = 3,
+        retry_delay: int = 10,
+        run_page_url: str | None = None,

Review Comment:
   Added `databricks_retry_args` too in hook initialization in the trigger  



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@airflow.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org