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

Question

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.

Answer 1

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