rawwar opened a new issue, #43127: URL: https://github.com/apache/airflow/issues/43127
### Description we can let users pass their own `_retryable_error` method, which will validate if retry is allowed or not. As of now, databricks operators only accept `retry_limit` and `retry_delay`. Adding support for custom retry function and after func should be good enough to handle all custom requirements. https://github.com/apache/airflow/blob/6b090b31038e6dd86f6efd49d81cf34b73d1e13e/providers/src/airflow/providers/databricks/hooks/databricks_base.py#L127 ### Use case/motivation It will allow user to set their own retry logics and also not require change to the provider everytime a user wants to retry for a particular type of exception. ### Related issues https://github.com/apache/airflow/issues/43080 ### Are you willing to submit a PR? - [X] Yes I am willing to submit a PR! ### Code of Conduct - [X] I agree to follow this project's [Code of Conduct](https://github.com/apache/airflow/blob/main/CODE_OF_CONDUCT.md) -- 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: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
