Re-add experimental flags (#34510)

* readd experimental flags

* black reformatting

Author: TajiHarrisMicrosoft

sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_deployment/batch/model_batch_deployment.py

@@ -14,6 +14,7 @@
 from azure.ai.ml._schema._deployment.deployment import DeploymentSchema
 from azure.ai.ml.constants._common import BASE_PATH_CONTEXT_KEY
 from azure.ai.ml.constants._deployment import BatchDeploymentType
+from azure.ai.ml._schema import ExperimentalField
 from .model_batch_deployment_settings import ModelBatchDeploymentSettingsSchema
 
 
@@ -36,7 +37,7 @@ class ModelBatchDeploymentSchema(DeploymentSchema):
         allowed_values=[BatchDeploymentType.PIPELINE, BatchDeploymentType.MODEL], required=False
     )
 
-    settings = NestedField(ModelBatchDeploymentSettingsSchema)
+    settings = ExperimentalField(NestedField(ModelBatchDeploymentSettingsSchema))
 
     @post_load
     def make(self, data: Any, **kwargs: Any) -> Any:

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_deployment/model_batch_deployment.py

@@ -12,6 +12,7 @@
 from azure.ai.ml._restclient.v2022_05_01.models import CodeConfiguration as RestCodeConfiguration
 from azure.ai.ml._restclient.v2022_05_01.models import IdAssetReference
 from azure.ai.ml._schema._deployment.batch.model_batch_deployment import ModelBatchDeploymentSchema
+from azure.ai.ml._utils._experimental import experimental
 from azure.ai.ml.constants._common import BASE_PATH_CONTEXT_KEY, PARAMS_OVERRIDE_KEY
 from azure.ai.ml.constants._deployment import BatchDeploymentOutputAction
 from azure.ai.ml.entities._assets import Environment, Model
@@ -25,6 +26,7 @@
 from .model_batch_deployment_settings import ModelBatchDeploymentSettings
 
 
+@experimental
 class ModelBatchDeployment(Deployment):
     """Job Definition entity.
 

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_deployment/model_batch_deployment_settings.py

@@ -5,11 +5,13 @@
 from typing import Any, Dict, Optional
 
 from azure.ai.ml._schema._deployment.batch.model_batch_deployment_settings import ModelBatchDeploymentSettingsSchema
+from azure.ai.ml._utils._experimental import experimental
 from azure.ai.ml.constants._common import BASE_PATH_CONTEXT_KEY
 from azure.ai.ml.constants._deployment import BatchDeploymentOutputAction
 from azure.ai.ml.entities._deployment.deployment_settings import BatchRetrySettings
 
 
+@experimental
 class ModelBatchDeploymentSettings:
     """Model Batch Deployment Settings entity.
 

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_deployment/pipeline_component_batch_deployment.py

@@ -16,6 +16,7 @@
     PipelineComponentBatchDeploymentSchema,
 )
 from azure.ai.ml._utils._arm_id_utils import _parse_endpoint_name_from_deployment_id
+from azure.ai.ml._utils._experimental import experimental
 from azure.ai.ml._utils.utils import dump_yaml_to_file
 from azure.ai.ml.constants._common import BASE_PATH_CONTEXT_KEY, PARAMS_OVERRIDE_KEY
 from azure.ai.ml.entities import PipelineComponent
@@ -25,6 +26,7 @@
 from azure.ai.ml.entities._util import load_from_dict
 
 
+@experimental
 class PipelineComponentBatchDeployment(Resource):
     """Pipeline Component Batch Deployment entity.