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

Add Airflow Orchestration Integration for Feature View Transformations #4695

Open
Vishnu-Rangiah opened this issue Oct 25, 2024 · 0 comments
Labels
kind/feature New feature or request

Comments

@Vishnu-Rangiah
Copy link

Vishnu-Rangiah commented Oct 25, 2024

Is your feature request related to a problem? Please describe.

Today users are not able to schedule their feature transformations using simple configuration provided to feast objects.

Describe the solution you'd like
The orchestration of feature pipelines should be handled by tools such as Airflow. Schedules can be provided using the @Transform decorator.

@transform(
    sources=[credit_data_batch],
    entities=[user],
    mode="python",
    batch_schedule=timedelta(days=1), # Will automatically create a DAG and deploy assets during CICD
    schema=[Field("user_id", String), Field("timestamp", Timestamp), Field("current_balance", Float64)],
)
def user_last_balance(transactions):
    return transactions[["user_id", "timestamp", "current_balance"]]

Ideally the user can define their feature logic and schedule using the feature_view / transform decorator and provide their Airflow instance (local, astronomer, composer, etc.) configuration via the feature_store.yaml. Then the CICD (feast apply) should validate and deploy the correct assets to handle the orchestration of the transformations through airflow.

Describe alternatives you've considered

  • Alternatively we could provide examples of how to create an Airflow DAG using the feast SDK to run transformation jobs and not have to worry about the scheduling element in the transform decorator

Additional context

@Vishnu-Rangiah Vishnu-Rangiah added the kind/feature New feature or request label Oct 25, 2024
@Vishnu-Rangiah Vishnu-Rangiah changed the title Add Airflow Orchestration support for Feature View transformations Support Airflow Orchestration for Feature View transformations Oct 25, 2024
@Vishnu-Rangiah Vishnu-Rangiah changed the title Support Airflow Orchestration for Feature View transformations Add Airflow Orchestration for Feature View transformations Oct 25, 2024
@Vishnu-Rangiah Vishnu-Rangiah changed the title Add Airflow Orchestration for Feature View transformations Add Airflow Orchestration Integration for Feature View transformations Oct 25, 2024
@Vishnu-Rangiah Vishnu-Rangiah changed the title Add Airflow Orchestration Integration for Feature View transformations Add Airflow Orchestration Integration for Feature View Transformations Oct 25, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
kind/feature New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant