使用 Azure Databricks Feature Store Client 在启用了统一目录的工作区上将功能发布到 cosmos dB

Question

我们正在尝试使用 cosmosdb 创建一个在线特征存储，遵循这个文档： https://learn.microsoft.com/en-us/azure/databricks/machine-learning/feature-store/publish-features 。

但是当我将表发布到 cosmosdb 时出现错误：AnalysisException: Catalog 'cosmoscatalog' not found。 该问题仅在使用支持统一的工作区时发生。 我可以使用未启用 unity 的工作区进行发布。

PS 如果我使用未启用 unity 的工作空间创建表，则启用 unity 的工作空间可以更新 cosmosdb。 但是启用 unity 的 worskpace 无法使用 fs.publish_table 创建 cosmos 容器/数据库。

我尝试了以下代码：

from databricks.feature_store.online_store_spec import AzureCosmosDBSpec
from databricks.feature_store.client import FeatureStoreClient

fs = FeatureStoreClient()

account_uri = "https://online-feature-store.documents.azure.com:443/"

# Specify the online store.
online_store_spec = AzureCosmosDBSpec(
  account_uri=account_uri,
  write_secret_prefix="secret/write-cosmos",
  read_secret_prefix="secret/read-cosmos",
  database_name="online_feature_store_example",
  container_name="feature_store_online_wine_features"
)

# Push the feature table to online store.
fs.publish_table("online_feature_store_example.wine_static_features", online_store_spec, mode='merge')

以下代码适用于未启用统一目录的工作区。 但是，在启用统一目录的工作区上，它会报错：AnalysisException: Catalog 'cosmoscatalog' not found

Answer 1

您需要使用您在 AzureCosmosDBSpec 中指定的名称在 CosmosDB 中创建数据库和容器。