Remove experimental flags from batch deployments (Azure#34216)

* remove experimental flags from batch inferencing entities and schemas for ga * remove unused import
LazaUK · Feb 8, 2024 · dbbae4e · dbbae4e
1 parent 81d03b4
commit dbbae4e
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 8 deletions.
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_deployment/batch/model_batch_deployment.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_deployment/batch/model_batch_deployment.py
@@ -14,7 +14,6 @@
 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
 
 
@@ -37,7 +36,7 @@ class ModelBatchDeploymentSchema(DeploymentSchema):
         allowed_values=[BatchDeploymentType.PIPELINE, BatchDeploymentType.MODEL], required=False
     )
 
-    settings = ExperimentalField(NestedField(ModelBatchDeploymentSettingsSchema))
+    settings = NestedField(ModelBatchDeploymentSettingsSchema)
 
     @post_load
     def make(self, data: Any, **kwargs: Any) -> Any:

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_deployment/model_batch_deployment.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_deployment/model_batch_deployment.py
@@ -12,7 +12,6 @@
 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
@@ -26,7 +25,6 @@
 from .model_batch_deployment_settings import ModelBatchDeploymentSettings
 
 
-@experimental
 class ModelBatchDeployment(Deployment):
     """Job Definition entity.
 

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_deployment/model_batch_deployment_settings.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_deployment/model_batch_deployment_settings.py
@@ -5,13 +5,11 @@
 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.
 

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_deployment/pipeline_component_batch_deployment.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_deployment/pipeline_component_batch_deployment.py
@@ -16,7 +16,6 @@
     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
@@ -26,7 +25,6 @@
 from azure.ai.ml.entities._util import load_from_dict
 
 
-@experimental
 class PipelineComponentBatchDeployment(Resource):
     """Pipeline Component Batch Deployment entity.