Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Integration test fix & Decorator change task.ray() -> ray.task() #35

Merged
merged 8 commits into from
Aug 7, 2024

Conversation

venkatajagannath
Copy link
Contributor

@venkatajagannath venkatajagannath commented Aug 7, 2024

This PR makes 2 changes

Decorator change

Using task.ray() in a DAG might requires us to provide an alias to base object task().

e.g:

from airflow.decorators import dag, task as airflow_task
from ray_provider.decorators.ray import task

Its better to not to have to give a different alias to the the main object in the taskflow api. So, I'm changing the code to be ray.task()

Integration test fix

We want to make sure our cluster name is unique if we are creating a GKE cluster. So, making changes to that portion.

Also, reducing the amount of resources allocated to the k8 cluster to control costs

@venkatajagannath venkatajagannath changed the title Decorator change task.ray() -> ray.task() Integration test fix & Decorator change task.ray() -> ray.task() Aug 7, 2024
@venkatajagannath venkatajagannath merged commit 738aa01 into main Aug 7, 2024
16 checks passed
@venkatajagannath venkatajagannath deleted the decorator-change branch August 7, 2024 06:35
@venkatajagannath venkatajagannath added bug Something isn't working enhancement New feature or request labels Aug 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant