[ML] Migrate from 2022_12_01_preview API to 2023_02_01_preview API (#…

…28863) * [ML] Migrate from 2022_12_01_preview API to 2023_02_01_preview API * Fix ml_client * Hardcode MFE url to master for testing the gates * TEMP change to sanitize MFE url * Automl * MLClient and Workspace * Workspace and Networking still depend on 2022-12 * ComputeRuntimeDTO and FeatureStoreSettings depends on v2022_12_01_preview * Move remaining workspace files to use 2022-12 API * Move workspace_keys files to use 2022-12 API * Move test_ws_operations to use 2022-12 API * Remove SAS URI Santizer * Remove hardcodes to use master mfe endpoint
Azure · Mar 3, 2023 · 36da8f0 · 36da8f0
1 parent bb3a543
commit 36da8f0
Show file tree

Hide file tree

Showing 104 changed files with 319 additions and 313 deletions.
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_internal/entities/command.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_internal/entities/command.py
@@ -10,8 +10,8 @@
 from azure.ai.ml._internal._schema.component import NodeType
 from azure.ai.ml._internal.entities.component import InternalComponent
 from azure.ai.ml._internal.entities.node import InternalBaseNode
-from azure.ai.ml._restclient.v2022_12_01_preview.models import CommandJobLimits as RestCommandJobLimits
-from azure.ai.ml._restclient.v2022_12_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
+from azure.ai.ml._restclient.v2023_02_01_preview.models import CommandJobLimits as RestCommandJobLimits
+from azure.ai.ml._restclient.v2023_02_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
 from azure.ai.ml._schema import PathAwareSchema
 from azure.ai.ml._schema.core.fields import DistributionField
 from azure.ai.ml.entities import CommandJobLimits, JobResourceConfiguration

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py
@@ -465,7 +465,7 @@ def __init__(
         self._jobs = JobOperations(
             self._operation_scope,
             self._operation_config,
-            self._service_client_12_2022_preview,
+            self._service_client_02_2023_preview,
             self._operation_container,
             self._credential,
             _service_client_kwargs=kwargs,

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/automl_vertical.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/automl_vertical.py
@@ -2,7 +2,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import LogVerbosity
+from azure.ai.ml._restclient.v2023_02_01_preview.models import LogVerbosity
 from azure.ai.ml._schema.automl.automl_job import AutoMLJobSchema
 from azure.ai.ml._schema.core.fields import NestedField, StringTransformedEnum, UnionField
 from azure.ai.ml._schema.job.input_output_entry import MLTableInputSchema

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/featurization_settings.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/featurization_settings.py
@@ -7,7 +7,7 @@
 from marshmallow import fields as flds
 from marshmallow import post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import BlockedTransformers
+from azure.ai.ml._restclient.v2023_02_01_preview.models import BlockedTransformers
 from azure.ai.ml._schema.core.fields import NestedField, StringTransformedEnum, UnionField
 from azure.ai.ml._schema.core.schema import PatchedSchemaMeta
 from azure.ai.ml._utils.utils import camel_to_snake

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/forecasting_settings.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/forecasting_settings.py
@@ -6,16 +6,16 @@
 
 from marshmallow import fields, post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import FeatureLags as FeatureLagsMode
-from azure.ai.ml._restclient.v2022_12_01_preview.models import (
+from azure.ai.ml._restclient.v2023_02_01_preview.models import FeatureLags as FeatureLagsMode
+from azure.ai.ml._restclient.v2023_02_01_preview.models import (
     ForecastHorizonMode,
     SeasonalityMode,
     ShortSeriesHandlingConfiguration,
     TargetAggregationFunction,
     TargetLagsMode,
     TargetRollingWindowSizeMode,
 )
-from azure.ai.ml._restclient.v2022_12_01_preview.models import UseStl as STLMode
+from azure.ai.ml._restclient.v2023_02_01_preview.models import UseStl as STLMode
 from azure.ai.ml._schema.core.fields import StringTransformedEnum, UnionField
 from azure.ai.ml._schema.core.schema import PatchedSchemaMeta
 

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_classification.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_classification.py
@@ -8,7 +8,7 @@
 
 from marshmallow import fields, post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import (
+from azure.ai.ml._restclient.v2023_02_01_preview.models import (
     ClassificationMultilabelPrimaryMetrics,
     ClassificationPrimaryMetrics,
     TaskType,

diff --git a/...zure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_model_distribution_settings.py b/...zure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_model_distribution_settings.py
@@ -6,7 +6,7 @@
 
 from marshmallow import fields, post_dump, post_load, pre_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import (
+from azure.ai.ml._restclient.v2023_02_01_preview.models import (
     LearningRateScheduler,
     ModelSize,
     StochasticOptimizer,

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_model_settings.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_model_settings.py
@@ -6,7 +6,7 @@
 
 from marshmallow import fields, post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import (
+from azure.ai.ml._restclient.v2023_02_01_preview.models import (
     LearningRateScheduler,
     ModelSize,
     StochasticOptimizer,

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_object_detection.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_object_detection.py
@@ -8,7 +8,7 @@
 
 from marshmallow import fields, post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import (
+from azure.ai.ml._restclient.v2023_02_01_preview.models import (
     InstanceSegmentationPrimaryMetrics,
     ObjectDetectionPrimaryMetrics,
     TaskType,

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/nlp_fixed_parameters.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/nlp_fixed_parameters.py
@@ -6,7 +6,7 @@
 
 from marshmallow import fields, post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import NlpLearningRateScheduler
+from azure.ai.ml._restclient.v2023_02_01_preview.models import NlpLearningRateScheduler
 from azure.ai.ml._schema.core.fields import StringTransformedEnum
 from azure.ai.ml._schema.core.schema import PatchedSchemaMeta
 from azure.ai.ml._utils.utils import camel_to_snake, snake_to_kebab

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/nlp_parameter_subspace.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/nlp_parameter_subspace.py
@@ -6,7 +6,7 @@
 
 from marshmallow import fields, post_dump, post_load, pre_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import NlpLearningRateScheduler
+from azure.ai.ml._restclient.v2023_02_01_preview.models import NlpLearningRateScheduler
 from azure.ai.ml._schema._sweep.search_space import (
     ChoiceSchema,
     NormalSchema,

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/text_classification.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/text_classification.py
@@ -8,7 +8,7 @@
 
 from marshmallow import post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import ClassificationPrimaryMetrics, TaskType
+from azure.ai.ml._restclient.v2023_02_01_preview.models import ClassificationPrimaryMetrics, TaskType
 from azure.ai.ml._schema.automl.nlp_vertical.nlp_vertical import NlpVerticalSchema
 from azure.ai.ml._schema.core.fields import StringTransformedEnum, fields
 from azure.ai.ml._utils.utils import camel_to_snake

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/text_classification_multilabel.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/text_classification_multilabel.py
@@ -8,7 +8,7 @@
 
 from marshmallow import post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import ClassificationMultilabelPrimaryMetrics, TaskType
+from azure.ai.ml._restclient.v2023_02_01_preview.models import ClassificationMultilabelPrimaryMetrics, TaskType
 from azure.ai.ml._schema.automl.nlp_vertical.nlp_vertical import NlpVerticalSchema
 from azure.ai.ml._schema.core.fields import StringTransformedEnum, fields
 from azure.ai.ml._utils.utils import camel_to_snake

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/text_ner.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/text_ner.py
@@ -8,7 +8,7 @@
 
 from marshmallow import post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import ClassificationPrimaryMetrics, TaskType
+from azure.ai.ml._restclient.v2023_02_01_preview.models import ClassificationPrimaryMetrics, TaskType
 from azure.ai.ml._schema.automl.nlp_vertical.nlp_vertical import NlpVerticalSchema
 from azure.ai.ml._schema.core.fields import StringTransformedEnum, fields
 from azure.ai.ml._utils.utils import camel_to_snake

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/classification.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/classification.py
@@ -8,7 +8,7 @@
 
 from marshmallow import fields, post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import ClassificationPrimaryMetrics, TaskType
+from azure.ai.ml._restclient.v2023_02_01_preview.models import ClassificationPrimaryMetrics, TaskType
 from azure.ai.ml._schema.automl.table_vertical.table_vertical import AutoMLTableVerticalSchema
 from azure.ai.ml._schema.automl.training_settings import ClassificationTrainingSettingsSchema
 from azure.ai.ml._schema.core.fields import NestedField, StringTransformedEnum

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/forecasting.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/forecasting.py
@@ -8,7 +8,7 @@
 
 from marshmallow import post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import ForecastingPrimaryMetrics, TaskType
+from azure.ai.ml._restclient.v2023_02_01_preview.models import ForecastingPrimaryMetrics, TaskType
 from azure.ai.ml._schema.automl.forecasting_settings import ForecastingSettingsSchema
 from azure.ai.ml._schema.automl.table_vertical.table_vertical import AutoMLTableVerticalSchema
 from azure.ai.ml._schema.automl.training_settings import ForecastingTrainingSettingsSchema

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/regression.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/regression.py
@@ -8,7 +8,7 @@
 
 from marshmallow import post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import RegressionPrimaryMetrics, TaskType
+from azure.ai.ml._restclient.v2023_02_01_preview.models import RegressionPrimaryMetrics, TaskType
 from azure.ai.ml._schema.automl.table_vertical.table_vertical import AutoMLTableVerticalSchema
 from azure.ai.ml._schema.automl.training_settings import RegressionTrainingSettingsSchema
 from azure.ai.ml._schema.core.fields import NestedField, StringTransformedEnum

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/table_vertical.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/table_vertical.py
@@ -4,7 +4,7 @@
 
 # pylint: disable=unused-argument,no-self-use
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import NCrossValidationsMode
+from azure.ai.ml._restclient.v2023_02_01_preview.models import NCrossValidationsMode
 from azure.ai.ml._schema.automl.automl_vertical import AutoMLVerticalSchema
 from azure.ai.ml._schema.automl.featurization_settings import TableFeaturizationSettingsSchema
 from azure.ai.ml._schema.automl.table_vertical.table_vertical_limit_settings import AutoMLTableLimitsSchema

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/training_settings.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/training_settings.py
@@ -6,7 +6,7 @@
 
 from marshmallow import fields, post_load
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import (
+from azure.ai.ml._restclient.v2023_02_01_preview.models import (
     ClassificationModels,
     ForecastingModels,
     RegressionModels,

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/automl/_automl_image.py b/sdk/ml/azure-ai-ml/azure/ai/ml/automl/_automl_image.py
@@ -6,7 +6,7 @@
 
 from typing import Optional, Union
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import (
+from azure.ai.ml._restclient.v2023_02_01_preview.models import (
     ClassificationMultilabelPrimaryMetrics,
     ClassificationPrimaryMetrics,
     InstanceSegmentationPrimaryMetrics,

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/command.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/command.py
@@ -13,10 +13,10 @@
 
 from marshmallow import INCLUDE, Schema
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import CommandJob as RestCommandJob
-from azure.ai.ml._restclient.v2022_12_01_preview.models import CommandJobLimits as RestCommandJobLimits
-from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase
-from azure.ai.ml._restclient.v2022_12_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
+from azure.ai.ml._restclient.v2023_02_01_preview.models import CommandJob as RestCommandJob
+from azure.ai.ml._restclient.v2023_02_01_preview.models import CommandJobLimits as RestCommandJobLimits
+from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase
+from azure.ai.ml._restclient.v2023_02_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
 from azure.ai.ml._schema.core.fields import NestedField, UnionField
 from azure.ai.ml._schema.job.command_job import CommandJobSchema
 from azure.ai.ml._schema.job.identity import AMLTokenIdentitySchema, ManagedIdentitySchema, UserIdentitySchema

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/parallel.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/parallel.py
@@ -13,7 +13,7 @@
 
 from marshmallow import Schema
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
+from azure.ai.ml._restclient.v2023_02_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
 from azure.ai.ml.constants._common import ARM_ID_PREFIX
 from azure.ai.ml.constants._component import NodeType
 from azure.ai.ml.entities._component.component import Component

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/spark.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/spark.py
@@ -12,11 +12,11 @@
 
 from marshmallow import INCLUDE, Schema
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import IdentityConfiguration
-from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase as JobBaseData
-from azure.ai.ml._restclient.v2022_12_01_preview.models import SparkJob as RestSparkJob
-from azure.ai.ml._restclient.v2022_12_01_preview.models import SparkJobEntry as RestSparkJobEntry
-from azure.ai.ml._restclient.v2022_12_01_preview.models import (
+from azure.ai.ml._restclient.v2023_02_01_preview.models import IdentityConfiguration
+from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase as JobBaseData
+from azure.ai.ml._restclient.v2023_02_01_preview.models import SparkJob as RestSparkJob
+from azure.ai.ml._restclient.v2023_02_01_preview.models import SparkJobEntry as RestSparkJobEntry
+from azure.ai.ml._restclient.v2023_02_01_preview.models import (
     SparkResourceConfiguration as RestSparkResourceConfiguration,
 )
 from azure.ai.ml._schema.job.identity import AMLTokenIdentitySchema, ManagedIdentitySchema, UserIdentitySchema

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/spark_func.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/spark_func.py
@@ -6,7 +6,7 @@
 import os
 from typing import Callable, Dict, List, Optional, Tuple, Union
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import AmlToken, ManagedIdentity, UserIdentity
+from azure.ai.ml._restclient.v2023_02_01_preview.models import AmlToken, ManagedIdentity, UserIdentity
 from azure.ai.ml.constants._common import AssetTypes
 from azure.ai.ml.constants._component import ComponentSource
 from azure.ai.ml.entities import Environment

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_credentials.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_credentials.py
@@ -43,12 +43,12 @@
     ServicePrincipalDatastoreSecrets as RestServicePrincipalDatastoreSecrets,
 )
 from azure.ai.ml._restclient.v2022_05_01.models import UserAssignedIdentity as RestUserAssignedIdentityConfiguration
-from azure.ai.ml._restclient.v2022_12_01_preview.models import AmlToken as RestAmlToken
-from azure.ai.ml._restclient.v2022_12_01_preview.models import IdentityConfiguration as RestJobIdentityConfiguration
-from azure.ai.ml._restclient.v2022_12_01_preview.models import IdentityConfigurationType
-from azure.ai.ml._restclient.v2022_12_01_preview.models import ManagedIdentity as RestJobManagedIdentity
-from azure.ai.ml._restclient.v2022_12_01_preview.models import ManagedServiceIdentity as RestRegistryManagedIdentity
-from azure.ai.ml._restclient.v2022_12_01_preview.models import UserIdentity as RestUserIdentity
+from azure.ai.ml._restclient.v2023_02_01_preview.models import AmlToken as RestAmlToken
+from azure.ai.ml._restclient.v2023_02_01_preview.models import IdentityConfiguration as RestJobIdentityConfiguration
+from azure.ai.ml._restclient.v2023_02_01_preview.models import IdentityConfigurationType
+from azure.ai.ml._restclient.v2023_02_01_preview.models import ManagedIdentity as RestJobManagedIdentity
+from azure.ai.ml._restclient.v2023_02_01_preview.models import ManagedServiceIdentity as RestRegistryManagedIdentity
+from azure.ai.ml._restclient.v2023_02_01_preview.models import UserIdentity as RestUserIdentity
 from azure.ai.ml._utils.utils import camel_to_snake, snake_to_pascal
 from azure.ai.ml.constants._common import CommonYamlFields, IdentityType
 from azure.ai.ml.entities._mixins import DictMixin, RestTranslatableMixin, YamlTranslatableMixin

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/_input_output_helpers.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/_input_output_helpers.py
@@ -6,24 +6,24 @@
 import re
 from typing import Any, Dict, Union
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import CustomModelJobInput as RestCustomModelJobInput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import CustomModelJobOutput as RestCustomModelJobOutput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import InputDeliveryMode
-from azure.ai.ml._restclient.v2022_12_01_preview.models import JobInput as RestJobInput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import JobInputType
-from azure.ai.ml._restclient.v2022_12_01_preview.models import JobOutput as RestJobOutput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import JobOutputType, LiteralJobInput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import MLFlowModelJobInput as RestMLFlowModelJobInput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import MLFlowModelJobOutput as RestMLFlowModelJobOutput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import MLTableJobInput as RestMLTableJobInput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import MLTableJobOutput as RestMLTableJobOutput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import OutputDeliveryMode
-from azure.ai.ml._restclient.v2022_12_01_preview.models import TritonModelJobInput as RestTritonModelJobInput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import TritonModelJobOutput as RestTritonModelJobOutput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import UriFileJobInput as RestUriFileJobInput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import UriFileJobOutput as RestUriFileJobOutput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import UriFolderJobInput as RestUriFolderJobInput
-from azure.ai.ml._restclient.v2022_12_01_preview.models import UriFolderJobOutput as RestUriFolderJobOutput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import CustomModelJobInput as RestCustomModelJobInput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import CustomModelJobOutput as RestCustomModelJobOutput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import InputDeliveryMode
+from azure.ai.ml._restclient.v2023_02_01_preview.models import JobInput as RestJobInput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import JobInputType
+from azure.ai.ml._restclient.v2023_02_01_preview.models import JobOutput as RestJobOutput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import JobOutputType, LiteralJobInput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import MLFlowModelJobInput as RestMLFlowModelJobInput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import MLFlowModelJobOutput as RestMLFlowModelJobOutput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import MLTableJobInput as RestMLTableJobInput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import MLTableJobOutput as RestMLTableJobOutput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import OutputDeliveryMode
+from azure.ai.ml._restclient.v2023_02_01_preview.models import TritonModelJobInput as RestTritonModelJobInput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import TritonModelJobOutput as RestTritonModelJobOutput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import UriFileJobInput as RestUriFileJobInput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import UriFileJobOutput as RestUriFileJobOutput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import UriFolderJobInput as RestUriFolderJobInput
+from azure.ai.ml._restclient.v2023_02_01_preview.models import UriFolderJobOutput as RestUriFolderJobOutput
 from azure.ai.ml._utils.utils import is_data_binding_expression
 from azure.ai.ml.constants import AssetTypes, InputOutputModes, JobType
 from azure.ai.ml.constants._component import IOConstants

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/automl/automl_job.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/automl/automl_job.py
@@ -8,7 +8,7 @@
 from abc import ABC, abstractmethod
 from typing import Any, Dict, Optional, Union
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase, MLTableJobInput, ResourceConfiguration, TaskType
+from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase, MLTableJobInput, ResourceConfiguration, TaskType
 from azure.ai.ml._utils.utils import camel_to_snake
 from azure.ai.ml.constants import JobType
 from azure.ai.ml.constants._common import TYPE, AssetTypes

diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/automl/image/automl_image.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/automl/image/automl_image.py
@@ -5,7 +5,7 @@
 from abc import ABC
 from typing import Dict, Optional, Union
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import LogVerbosity, SamplingAlgorithmType
+from azure.ai.ml._restclient.v2023_02_01_preview.models import LogVerbosity, SamplingAlgorithmType
 from azure.ai.ml._utils.utils import camel_to_snake
 from azure.ai.ml.entities._inputs_outputs import Input
 from azure.ai.ml.entities._job.automl.automl_vertical import AutoMLVertical

diff --git a/...ml/azure-ai-ml/azure/ai/ml/entities/_job/automl/image/automl_image_classification_base.py b/...ml/azure-ai-ml/azure/ai/ml/entities/_job/automl/image/automl_image_classification_base.py
@@ -6,7 +6,7 @@
 
 from typing import Dict, List, Optional, Union
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import LearningRateScheduler, StochasticOptimizer
+from azure.ai.ml._restclient.v2023_02_01_preview.models import LearningRateScheduler, StochasticOptimizer
 from azure.ai.ml._utils.utils import camel_to_snake
 from azure.ai.ml.entities._job.automl.image.automl_image import AutoMLImage
 from azure.ai.ml.entities._job.automl.image.image_classification_search_space import ImageClassificationSearchSpace

diff --git a/.../azure-ai-ml/azure/ai/ml/entities/_job/automl/image/automl_image_object_detection_base.py b/.../azure-ai-ml/azure/ai/ml/entities/_job/automl/image/automl_image_object_detection_base.py
@@ -6,7 +6,7 @@
 
 from typing import Dict, List, Optional, Union
 
-from azure.ai.ml._restclient.v2022_12_01_preview.models import (
+from azure.ai.ml._restclient.v2023_02_01_preview.models import (
     LearningRateScheduler,
     ModelSize,
     StochasticOptimizer,