pythontypeerrordatabrickswebhooksmlflow

How do I resolve the TypeError with databricks.proto for databricks-registry-webhooks?

I am trying to create webhooks for MLflow in Databricks. However, I am getting the following TypeError when importing from databricks_registry_webhooks import RegistryWebhooksClient, JobSpec, HttpUrlSpec

TypeError: Couldn't build proto file into descriptor pool: duplicate file name databricks.proto

It is the same issue I found on the community site here.

Solution

  • The issue is that databricks-registry-webhooks has a databricks.proto file that collides with mlflow. Here is the fix:

    %pip install databricks-registry-webhooks mlflow==2.2.2

    You can test this is successful by running:

     from databricks_registry_webhooks import RegistryWebhooksClient, JobSpec, HttpUrlSpec