Closed niall-turbitt closed 1 year ago
Amazing. This is such a huge change. Great work and many thanks.
@arpitjasa-db made some changes to get Feature Store working with models in UC
Example AWS repo with Feature Store and models in UC: https://github.com/databricks/example-aws-github-project-fs-uc
Example AWS repo with no Feature Store and models in UC: https://github.com/databricks/example-aws-github-project-uc
Models in UC addition to MLOps Stack
Additional prompts added for project instantiation:
include_models_in_unity_catalog
: Whether to use Models in UC for model registration. Creates resources using DABs. Creates models under the catalog, mapped to the target environment. For example, if the bundle is deployed to prod, the model will be registered to the prod catalog. We assumed there exists a catalog mapped to each of these. Given thattest
is also a defined target, this assumes the existence of a test catalog if deploying to thetest
environmentschema_name
: Name of the schema under each catalog to register models to. Assumes the same schema name for each catalog. Defaults to "schema_name"unity_catalog_read_user_group
: Used to define thegrants
block assigned to the registered model in UC. Default to "account users"Notably, changes to accommodate models in UC were made to:
NOTE:
Testing Example AWS repo with Feature Store and models in UC: https://github.com/databricks/example-aws-github-project-fs-uc
Example AWS repo with no Feature Store and models in UC: https://github.com/databricks/example-aws-github-project-uc