Add databricks deployments client skeleton + example

examples/deployments/databricks.py

@@ -0,0 +1,15 @@
+from mlflow.deployments import get_deploy_client
+
+
+def main():
+    client = get_deploy_client("databricks")
+    client.create_endpoint(
+        name="gpt4-chat",
+        config={
+            # TODO: doesn't work yet
+        },
+    )
+
+
+if __name__ == "__main__":
+    main()

mlflow/deployments/databricks/__init__.py

@@ -0,0 +1,44 @@
+from mlflow.deployments import BaseDeploymentClient
+
+
+class DatabricksDeploymentClient(BaseDeploymentClient):
+    def create_deployment(self, name, model_uri, flavor=None, config=None, endpoint=None):
+        raise NotImplementedError
+
+    def update_deployment(self, name, model_uri=None, flavor=None, config=None, endpoint=None):
+        raise NotImplementedError
+
+    def delete_deployment(self, name, config=None, endpoint=None):
+        raise NotImplementedError
+
+    def list_deployments(self, endpoint=None):
+        raise NotImplementedError
+
+    def get_deployment(self, name, endpoint=None):
+        raise NotImplementedError
+
+    def predict(self, deployment_name=None, inputs=None, endpoint=None):
+        raise NotImplementedError("TODO")
+
+    def create_endpoint(self, name, config=None):
+        raise NotImplementedError("TODO")
+
+    def update_endpoint(self, endpoint, config=None):
+        raise NotImplementedError("TODO")
+
+    def delete_endpoint(self, endpoint):
+        raise NotImplementedError("TODO")
+
+    def list_endpoints(self):
+        raise NotImplementedError("TODO")
+
+    def get_endpoint(self, endpoint):
+        raise NotImplementedError("TODO")
+
+
+def run_local(name, model_uri, flavor=None, config=None):
+    pass
+
+
+def target_help():
+    pass

mlflow/gateway/fluent.py

@@ -1,4 +1,3 @@
-import logging
 from typing import Any, Dict, List, Optional
 
 from mlflow.gateway.client import MlflowGatewayClient

setup.py

@@ -184,6 +184,9 @@ def run(self):
 
         [mlflow.app.client]
         basic-auth=mlflow.server.auth.client:AuthServiceClient
+
+        [mlflow.deployments]
+        databricks=mlflow.deployments.databricks
     """,
     cmdclass={
         "dependencies": ListDependencies,