w.feature_store: Feature Store

class databricks.sdk.service.ml.FeatureStoreAPI

A feature store is a centralized repository that enables data scientists to find and share features. Using a feature store also ensures that the code used to compute feature values is the same during model training and when the model is used for inference.

An online store is a low-latency database used for feature lookup during real-time model inference or serve feature for real-time applications.

create_online_store(online_store: OnlineStore) → OnlineStore

Create an Online Feature Store.

Parameters:

online_store – OnlineStore Online store to create.

Returns:

OnlineStore

delete_online_store(name: str)

Delete an Online Feature Store.

Parameters:

name – str Name of the online store to delete.

delete_online_table(online_table_name: str)

Delete online table.

Parameters:

online_table_name – str The full three-part (catalog, schema, table) name of the online table.

get_online_store(name: str) → OnlineStore

Get an Online Feature Store.

Parameters:

name – str Name of the online store to get.

Returns:

OnlineStore

list_online_stores([, page_size: Optional[int], page_token: Optional[str]]) → Iterator[OnlineStore]

List Online Feature Stores.

Parameters:

  • page_size – int (optional) The maximum number of results to return. Defaults to 100 if not specified.

  • page_token – str (optional) Pagination token to go to the next page based on a previous query.

Returns:

Iterator over OnlineStore

publish_table(source_table_name: str, publish_spec: PublishSpec) → PublishTableResponse

Publish features.

Parameters:

  • source_table_name – str The full three-part (catalog, schema, table) name of the source table.

  • publish_spec – PublishSpec The specification for publishing the online table from the source table.

Returns:

PublishTableResponse

update_online_store(name: str, online_store: OnlineStore, update_mask: str) → OnlineStore

Update an Online Feature Store.

Parameters:

  • name – str The name of the online store. This is the unique identifier for the online store.

  • online_store – OnlineStore Online store to update.

  • update_mask – str The list of fields to update.

Returns:

OnlineStore