This PR adds Databricks Feature Store support as a component in MLOps Stacks. It follows similar principles applied to training code support, including sample feature engineering code, automated unit and integration testing and infra-as-code setup for deployment.
It includes the following:
Sample feature code (transform) modules.
Parameterized ETL notebooks that load data, apply feature modules and write toe Feature Store tables.
Infra as code terraform modules to deploy above ETL notebooks as jobs.
Sample model training code that is integrated with feature store client libraries.
Automated (via Github Workflows) integration tests that trigger feature engineering pipelines and training code and post results to Github UI as PR comments.
This PR adds Databricks Feature Store support as a component in MLOps Stacks. It follows similar principles applied to training code support, including sample feature engineering code, automated unit and integration testing and infra-as-code setup for deployment.
It includes the following:
The sample code is based on the Feature Store Taxi Example Notebook.
This functionality is optional and guarded by a cookiecutter boolean flag (
include_feature_store
).Tested:
TODOs (fixed):