astronomer / astro-provider-ray

This provider contains operators, decorators and triggers to send a ray job from an airflow task
https://astronomer.github.io/astro-provider-ray/
Apache License 2.0
12 stars 2 forks source link

Make max_retries, retry_interval param of _wait_for_load_balancer configurable #86

Open pankajastro opened 2 weeks ago

pankajastro commented 2 weeks ago

_wait_for_load_balancer may take some time longer so it may make sense to give an option to configure max_retries, retry_interval

https://github.com/astronomer/astro-provider-ray/blob/0d99128e1739192c38d8cdb8906e5b67970ce2c2/ray_provider/hooks/ray.py#L320-L321