From 12f22798b7dbe1a353913de62437db4f70509e12 Mon Sep 17 00:00:00 2001
From: Harutaka Kawamura <hkawamura0130@gmail.com>
Date: Tue, 21 Nov 2023 14:49:41 +0900
Subject: [PATCH] Mark deployment clients as experimental (#10468)

Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
---
 mlflow/deployments/databricks/__init__.py | 8 ++++++++
 mlflow/deployments/mlflow/__init__.py     | 5 +++++
 2 files changed, 13 insertions(+)

diff --git a/mlflow/deployments/databricks/__init__.py b/mlflow/deployments/databricks/__init__.py
index c52226b273a55..67c5f85c242df 100644
--- a/mlflow/deployments/databricks/__init__.py
+++ b/mlflow/deployments/databricks/__init__.py
@@ -8,6 +8,7 @@
 )
 from mlflow.environment_variables import MLFLOW_HTTP_REQUEST_TIMEOUT
 from mlflow.utils import AttrDict
+from mlflow.utils.annotations import experimental
 from mlflow.utils.databricks_utils import get_databricks_host_creds
 from mlflow.utils.rest_utils import augmented_raise_for_status, http_request
 
@@ -16,6 +17,7 @@ class DatabricksEndpoint(AttrDict):
     pass
 
 
+@experimental
 class DatabricksDeploymentClient(BaseDeploymentClient):
     """
     TODO
@@ -88,6 +90,7 @@ def _call_endpoint(
         augmented_raise_for_status(response)
         return DatabricksEndpoint(response.json())
 
+    @experimental
     def predict(self, deployment_name=None, inputs=None, endpoint=None):
         """
         TODO
@@ -100,6 +103,7 @@ def predict(self, deployment_name=None, inputs=None, endpoint=None):
             timeout=MLFLOW_DEPLOYMENT_PREDICT_TIMEOUT.get(),
         )
 
+    @experimental
     def create_endpoint(self, name, config=None):
         """
         TODO
@@ -112,6 +116,7 @@ def create_endpoint(self, name, config=None):
         payload = {"name": name, "config": config, **extras}
         return self._call_endpoint(method="POST", json_body=payload)
 
+    @experimental
     def update_endpoint(self, endpoint, config=None):
         """
         TODO
@@ -120,18 +125,21 @@ def update_endpoint(self, endpoint, config=None):
             method="PUT", route=posixpath.join(endpoint, "config"), json_body=config
         )
 
+    @experimental
     def delete_endpoint(self, endpoint):
         """
         TODO
         """
         return self._call_endpoint(method="DELETE", route=endpoint)
 
+    @experimental
     def list_endpoints(self):
         """
         TODO
         """
         return self._call_endpoint(method="GET").endpoints
 
+    @experimental
     def get_endpoint(self, endpoint):
         """
         TODO
diff --git a/mlflow/deployments/mlflow/__init__.py b/mlflow/deployments/mlflow/__init__.py
index 3c2f827fe3036..61553fa7dbb2a 100644
--- a/mlflow/deployments/mlflow/__init__.py
+++ b/mlflow/deployments/mlflow/__init__.py
@@ -1,6 +1,8 @@
 from mlflow.deployments import BaseDeploymentClient
+from mlflow.utils.annotations import experimental
 
 
+@experimental
 class MLflowDeploymentClient(BaseDeploymentClient):
     """
     TODO
@@ -61,18 +63,21 @@ def delete_endpoint(self, endpoint):
         """
         raise NotImplementedError
 
+    @experimental
     def predict(self, deployment_name=None, inputs=None, endpoint=None):
         """
         TODO
         """
         raise NotImplementedError("TODO")
 
+    @experimental
     def list_endpoints(self):
         """
         TODO
         """
         raise NotImplementedError("TODO")
 
+    @experimental
     def get_endpoint(self, endpoint):
         """
         TODO