diff --git a/sdk/ml/azure-ai-ml/CHANGELOG.md b/sdk/ml/azure-ai-ml/CHANGELOG.md
index bafbaa6393a4..1a159d21c89a 100644
--- a/sdk/ml/azure-ai-ml/CHANGELOG.md
+++ b/sdk/ml/azure-ai-ml/CHANGELOG.md
@@ -7,6 +7,7 @@
 - Workspace Connection has new read-only subclass: AzureBlobStoreWorkspaceConnectionSchema.
 - Workspace Connection supports 2 new types under main class: gen 2 and azure_one_lake.
 - Workspace Connection LIST operation can return data connections via new optional flag: include_data_connections.
+- Support `ml_client.schedules.trigger(name='my_schedule')` function to trigger a schedule once.
 
 ### Bugs Fixed
 - Fix pipeline job `outputs` not load correctly when `component: <local-file>` exists in pipeline job yaml.
diff --git a/sdk/ml/azure-ai-ml/assets.json b/sdk/ml/azure-ai-ml/assets.json
index e62f66b528d9..b83d09c5d195 100644
--- a/sdk/ml/azure-ai-ml/assets.json
+++ b/sdk/ml/azure-ai-ml/assets.json
@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/ml/azure-ai-ml",
-  "Tag": "python/ml/azure-ai-ml_67cc2fc9de"
+  "Tag": "python/ml/azure-ai-ml_1af7fe2f76"
 }
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
index 0e3b3c7242ad..86ac9f1ee413 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py
@@ -345,6 +345,13 @@ def __init__(
             **kwargs,
         )
 
+        self._service_client_01_2024_preview = ServiceClient012024Preview(
+            credential=self._credential,
+            subscription_id=self._operation_scope._subscription_id,
+            base_url=base_url,
+            **kwargs,
+        )
+
         # A general purpose, user-configurable pipeline for making
         # http requests
         self._requests_pipeline = HttpPipeline(**kwargs)
@@ -566,7 +573,7 @@ def __init__(
         self._components = ComponentOperations(
             self._operation_scope,
             self._operation_config,
-            self._service_client_10_2021_dataplanepreview if registry_name else self._service_client_10_2022,
+            self._service_client_10_2021_dataplanepreview if registry_name else self._service_client_01_2024_preview,
             self._operation_container,
             self._preflight,
             **ops_kwargs,
@@ -580,7 +587,7 @@ def __init__(
             self._credential,
             _service_client_kwargs=kwargs,
             requests_pipeline=self._requests_pipeline,
-            service_client_08_2023_preview=self._service_client_08_2023_preview,
+            service_client_01_2024_preview=self._service_client_01_2024_preview,
             **ops_kwargs,
         )
         self._operation_container.add(AzureMLResourceType.JOB, self._jobs)
@@ -588,6 +595,7 @@ def __init__(
             self._operation_scope,
             self._operation_config,
             self._service_client_06_2023_preview,
+            self._service_client_01_2024_preview,
             self._operation_container,
             self._credential,
             _service_client_kwargs=kwargs,
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/_azure_machine_learning_workspaces.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/_azure_machine_learning_workspaces.py
index 1b124a2eb5ae..7e0808cc8c02 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/_azure_machine_learning_workspaces.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/_azure_machine_learning_workspaces.py
@@ -15,7 +15,63 @@
 
 from . import models
 from ._configuration import AzureMachineLearningWorkspacesConfiguration
-from .operations import BatchDeploymentsOperations, BatchEndpointsOperations, CapacityReservationGroupsOperations, CodeContainersOperations, CodeVersionsOperations, ComponentContainersOperations, ComponentVersionsOperations, ComputeOperations, DataContainersOperations, DataVersionsOperations, DatastoresOperations, EndpointDeploymentOperations, EndpointOperations, EnvironmentContainersOperations, EnvironmentVersionsOperations, FeaturesOperations, FeaturesetContainersOperations, FeaturesetVersionsOperations, FeaturestoreEntityContainersOperations, FeaturestoreEntityVersionsOperations, InferenceEndpointsOperations, InferenceGroupsOperations, InferencePoolsOperations, JobsOperations, LabelingJobsOperations, ManagedNetworkProvisionsOperations, ManagedNetworkSettingsRuleOperations, MarketplaceSubscriptionsOperations, ModelContainersOperations, ModelVersionsOperations, OnlineDeploymentsOperations, OnlineEndpointsOperations, Operations, PrivateEndpointConnectionsOperations, PrivateLinkResourcesOperations, QuotasOperations, RegistriesOperations, RegistryCodeContainersOperations, RegistryCodeVersionsOperations, RegistryComponentContainersOperations, RegistryComponentVersionsOperations, RegistryDataContainersOperations, RegistryDataReferencesOperations, RegistryDataVersionsOperations, RegistryEnvironmentContainersOperations, RegistryEnvironmentVersionsOperations, RegistryModelContainersOperations, RegistryModelVersionsOperations, SchedulesOperations, ServerlessEndpointsOperations, UsagesOperations, VirtualMachineSizesOperations, WorkspaceConnectionsOperations, WorkspaceFeaturesOperations, WorkspacesOperations
+from .operations import (
+    BatchDeploymentsOperations,
+    BatchEndpointsOperations,
+    CapacityReservationGroupsOperations,
+    CodeContainersOperations,
+    CodeVersionsOperations,
+    ComponentContainersOperations,
+    ComponentVersionsOperations,
+    ComputeOperations,
+    DataContainersOperations,
+    DatastoresOperations,
+    DataVersionsOperations,
+    EndpointDeploymentOperations,
+    EndpointOperations,
+    EnvironmentContainersOperations,
+    EnvironmentVersionsOperations,
+    FeaturesetContainersOperations,
+    FeaturesetVersionsOperations,
+    FeaturesOperations,
+    FeaturestoreEntityContainersOperations,
+    FeaturestoreEntityVersionsOperations,
+    InferenceEndpointsOperations,
+    InferenceGroupsOperations,
+    InferencePoolsOperations,
+    JobsOperations,
+    LabelingJobsOperations,
+    ManagedNetworkProvisionsOperations,
+    ManagedNetworkSettingsRuleOperations,
+    MarketplaceSubscriptionsOperations,
+    ModelContainersOperations,
+    ModelVersionsOperations,
+    OnlineDeploymentsOperations,
+    OnlineEndpointsOperations,
+    Operations,
+    PrivateEndpointConnectionsOperations,
+    PrivateLinkResourcesOperations,
+    QuotasOperations,
+    RegistriesOperations,
+    RegistryCodeContainersOperations,
+    RegistryCodeVersionsOperations,
+    RegistryComponentContainersOperations,
+    RegistryComponentVersionsOperations,
+    RegistryDataContainersOperations,
+    RegistryDataReferencesOperations,
+    RegistryDataVersionsOperations,
+    RegistryEnvironmentContainersOperations,
+    RegistryEnvironmentVersionsOperations,
+    RegistryModelContainersOperations,
+    RegistryModelVersionsOperations,
+    SchedulesOperations,
+    ServerlessEndpointsOperations,
+    UsagesOperations,
+    VirtualMachineSizesOperations,
+    WorkspaceConnectionsOperations,
+    WorkspaceFeaturesOperations,
+    WorkspacesOperations,
+)
 
 if TYPE_CHECKING:
     # pylint: disable=unused-import,ungrouped-imports
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/_azure_machine_learning_workspaces.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/_azure_machine_learning_workspaces.py
index d0c96df6fd2e..d26d767abb06 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/_azure_machine_learning_workspaces.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/_azure_machine_learning_workspaces.py
@@ -7,7 +7,7 @@
 # --------------------------------------------------------------------------
 
 from copy import deepcopy
-from typing import Any, Awaitable, TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, Awaitable
 
 from msrest import Deserializer, Serializer
 
@@ -16,7 +16,63 @@
 
 from .. import models
 from ._configuration import AzureMachineLearningWorkspacesConfiguration
-from .operations import BatchDeploymentsOperations, BatchEndpointsOperations, CapacityReservationGroupsOperations, CodeContainersOperations, CodeVersionsOperations, ComponentContainersOperations, ComponentVersionsOperations, ComputeOperations, DataContainersOperations, DataVersionsOperations, DatastoresOperations, EndpointDeploymentOperations, EndpointOperations, EnvironmentContainersOperations, EnvironmentVersionsOperations, FeaturesOperations, FeaturesetContainersOperations, FeaturesetVersionsOperations, FeaturestoreEntityContainersOperations, FeaturestoreEntityVersionsOperations, InferenceEndpointsOperations, InferenceGroupsOperations, InferencePoolsOperations, JobsOperations, LabelingJobsOperations, ManagedNetworkProvisionsOperations, ManagedNetworkSettingsRuleOperations, MarketplaceSubscriptionsOperations, ModelContainersOperations, ModelVersionsOperations, OnlineDeploymentsOperations, OnlineEndpointsOperations, Operations, PrivateEndpointConnectionsOperations, PrivateLinkResourcesOperations, QuotasOperations, RegistriesOperations, RegistryCodeContainersOperations, RegistryCodeVersionsOperations, RegistryComponentContainersOperations, RegistryComponentVersionsOperations, RegistryDataContainersOperations, RegistryDataReferencesOperations, RegistryDataVersionsOperations, RegistryEnvironmentContainersOperations, RegistryEnvironmentVersionsOperations, RegistryModelContainersOperations, RegistryModelVersionsOperations, SchedulesOperations, ServerlessEndpointsOperations, UsagesOperations, VirtualMachineSizesOperations, WorkspaceConnectionsOperations, WorkspaceFeaturesOperations, WorkspacesOperations
+from .operations import (
+    BatchDeploymentsOperations,
+    BatchEndpointsOperations,
+    CapacityReservationGroupsOperations,
+    CodeContainersOperations,
+    CodeVersionsOperations,
+    ComponentContainersOperations,
+    ComponentVersionsOperations,
+    ComputeOperations,
+    DataContainersOperations,
+    DatastoresOperations,
+    DataVersionsOperations,
+    EndpointDeploymentOperations,
+    EndpointOperations,
+    EnvironmentContainersOperations,
+    EnvironmentVersionsOperations,
+    FeaturesetContainersOperations,
+    FeaturesetVersionsOperations,
+    FeaturesOperations,
+    FeaturestoreEntityContainersOperations,
+    FeaturestoreEntityVersionsOperations,
+    InferenceEndpointsOperations,
+    InferenceGroupsOperations,
+    InferencePoolsOperations,
+    JobsOperations,
+    LabelingJobsOperations,
+    ManagedNetworkProvisionsOperations,
+    ManagedNetworkSettingsRuleOperations,
+    MarketplaceSubscriptionsOperations,
+    ModelContainersOperations,
+    ModelVersionsOperations,
+    OnlineDeploymentsOperations,
+    OnlineEndpointsOperations,
+    Operations,
+    PrivateEndpointConnectionsOperations,
+    PrivateLinkResourcesOperations,
+    QuotasOperations,
+    RegistriesOperations,
+    RegistryCodeContainersOperations,
+    RegistryCodeVersionsOperations,
+    RegistryComponentContainersOperations,
+    RegistryComponentVersionsOperations,
+    RegistryDataContainersOperations,
+    RegistryDataReferencesOperations,
+    RegistryDataVersionsOperations,
+    RegistryEnvironmentContainersOperations,
+    RegistryEnvironmentVersionsOperations,
+    RegistryModelContainersOperations,
+    RegistryModelVersionsOperations,
+    SchedulesOperations,
+    ServerlessEndpointsOperations,
+    UsagesOperations,
+    VirtualMachineSizesOperations,
+    WorkspaceConnectionsOperations,
+    WorkspaceFeaturesOperations,
+    WorkspacesOperations,
+)
 
 if TYPE_CHECKING:
     # pylint: disable=unused-import,ungrouped-imports
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/__init__.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/__init__.py
index aefdb5620dd3..ed0728892746 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/__init__.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/__init__.py
@@ -6,61 +6,61 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-from ._usages_operations import UsagesOperations
-from ._virtual_machine_sizes_operations import VirtualMachineSizesOperations
-from ._quotas_operations import QuotasOperations
-from ._compute_operations import ComputeOperations
-from ._registries_operations import RegistriesOperations
-from ._workspace_features_operations import WorkspaceFeaturesOperations
-from ._capacity_reservation_groups_operations import CapacityReservationGroupsOperations
-from ._registry_code_containers_operations import RegistryCodeContainersOperations
-from ._registry_code_versions_operations import RegistryCodeVersionsOperations
-from ._registry_component_containers_operations import RegistryComponentContainersOperations
-from ._registry_component_versions_operations import RegistryComponentVersionsOperations
-from ._registry_data_containers_operations import RegistryDataContainersOperations
-from ._registry_data_versions_operations import RegistryDataVersionsOperations
-from ._registry_data_references_operations import RegistryDataReferencesOperations
-from ._registry_environment_containers_operations import RegistryEnvironmentContainersOperations
-from ._registry_environment_versions_operations import RegistryEnvironmentVersionsOperations
-from ._marketplace_subscriptions_operations import MarketplaceSubscriptionsOperations
-from ._registry_model_containers_operations import RegistryModelContainersOperations
-from ._registry_model_versions_operations import RegistryModelVersionsOperations
-from ._batch_endpoints_operations import BatchEndpointsOperations
 from ._batch_deployments_operations import BatchDeploymentsOperations
+from ._batch_endpoints_operations import BatchEndpointsOperations
+from ._capacity_reservation_groups_operations import CapacityReservationGroupsOperations
 from ._code_containers_operations import CodeContainersOperations
 from ._code_versions_operations import CodeVersionsOperations
 from ._component_containers_operations import ComponentContainersOperations
 from ._component_versions_operations import ComponentVersionsOperations
+from ._compute_operations import ComputeOperations
 from ._data_containers_operations import DataContainersOperations
 from ._data_versions_operations import DataVersionsOperations
 from ._datastores_operations import DatastoresOperations
+from ._endpoint_deployment_operations import EndpointDeploymentOperations
+from ._endpoint_operations import EndpointOperations
 from ._environment_containers_operations import EnvironmentContainersOperations
 from ._environment_versions_operations import EnvironmentVersionsOperations
-from ._featureset_containers_operations import FeaturesetContainersOperations
 from ._features_operations import FeaturesOperations
+from ._featureset_containers_operations import FeaturesetContainersOperations
 from ._featureset_versions_operations import FeaturesetVersionsOperations
 from ._featurestore_entity_containers_operations import FeaturestoreEntityContainersOperations
 from ._featurestore_entity_versions_operations import FeaturestoreEntityVersionsOperations
-from ._inference_pools_operations import InferencePoolsOperations
 from ._inference_endpoints_operations import InferenceEndpointsOperations
 from ._inference_groups_operations import InferenceGroupsOperations
+from ._inference_pools_operations import InferencePoolsOperations
 from ._jobs_operations import JobsOperations
 from ._labeling_jobs_operations import LabelingJobsOperations
+from ._managed_network_provisions_operations import ManagedNetworkProvisionsOperations
+from ._managed_network_settings_rule_operations import ManagedNetworkSettingsRuleOperations
+from ._marketplace_subscriptions_operations import MarketplaceSubscriptionsOperations
 from ._model_containers_operations import ModelContainersOperations
 from ._model_versions_operations import ModelVersionsOperations
-from ._online_endpoints_operations import OnlineEndpointsOperations
 from ._online_deployments_operations import OnlineDeploymentsOperations
-from ._schedules_operations import SchedulesOperations
-from ._serverless_endpoints_operations import ServerlessEndpointsOperations
+from ._online_endpoints_operations import OnlineEndpointsOperations
 from ._operations import Operations
-from ._workspaces_operations import WorkspacesOperations
-from ._workspace_connections_operations import WorkspaceConnectionsOperations
-from ._endpoint_deployment_operations import EndpointDeploymentOperations
-from ._endpoint_operations import EndpointOperations
-from ._managed_network_settings_rule_operations import ManagedNetworkSettingsRuleOperations
 from ._private_endpoint_connections_operations import PrivateEndpointConnectionsOperations
 from ._private_link_resources_operations import PrivateLinkResourcesOperations
-from ._managed_network_provisions_operations import ManagedNetworkProvisionsOperations
+from ._quotas_operations import QuotasOperations
+from ._registries_operations import RegistriesOperations
+from ._registry_code_containers_operations import RegistryCodeContainersOperations
+from ._registry_code_versions_operations import RegistryCodeVersionsOperations
+from ._registry_component_containers_operations import RegistryComponentContainersOperations
+from ._registry_component_versions_operations import RegistryComponentVersionsOperations
+from ._registry_data_containers_operations import RegistryDataContainersOperations
+from ._registry_data_references_operations import RegistryDataReferencesOperations
+from ._registry_data_versions_operations import RegistryDataVersionsOperations
+from ._registry_environment_containers_operations import RegistryEnvironmentContainersOperations
+from ._registry_environment_versions_operations import RegistryEnvironmentVersionsOperations
+from ._registry_model_containers_operations import RegistryModelContainersOperations
+from ._registry_model_versions_operations import RegistryModelVersionsOperations
+from ._schedules_operations import SchedulesOperations
+from ._serverless_endpoints_operations import ServerlessEndpointsOperations
+from ._usages_operations import UsagesOperations
+from ._virtual_machine_sizes_operations import VirtualMachineSizesOperations
+from ._workspace_connections_operations import WorkspaceConnectionsOperations
+from ._workspace_features_operations import WorkspaceFeaturesOperations
+from ._workspaces_operations import WorkspacesOperations
 
 __all__ = [
     'UsagesOperations',
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_code_versions_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_code_versions_operations.py
index 9e432615f87e..e691f2f4c64d 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_code_versions_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_code_versions_operations.py
@@ -9,7 +9,13 @@
 from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar, Union
 
 from azure.core.async_paging import AsyncItemPaged, AsyncList
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import AsyncHttpResponse
 from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
@@ -21,7 +27,15 @@
 
 from ... import models as _models
 from ..._vendor import _convert_request
-from ...operations._code_versions_operations import build_create_or_get_start_pending_upload_request, build_create_or_update_request, build_delete_request, build_get_request, build_list_request, build_publish_request_initial
+from ...operations._code_versions_operations import (
+    build_create_or_get_start_pending_upload_request,
+    build_create_or_update_request,
+    build_delete_request,
+    build_get_request,
+    build_list_request,
+    build_publish_request_initial,
+)
+
 T = TypeVar('T')
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
 
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_component_versions_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_component_versions_operations.py
index 2ae0c49399a7..ca8e15b3c1e2 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_component_versions_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_component_versions_operations.py
@@ -9,7 +9,13 @@
 from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar, Union
 
 from azure.core.async_paging import AsyncItemPaged, AsyncList
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import AsyncHttpResponse
 from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
@@ -21,7 +27,14 @@
 
 from ... import models as _models
 from ..._vendor import _convert_request
-from ...operations._component_versions_operations import build_create_or_update_request, build_delete_request, build_get_request, build_list_request, build_publish_request_initial
+from ...operations._component_versions_operations import (
+    build_create_or_update_request,
+    build_delete_request,
+    build_get_request,
+    build_list_request,
+    build_publish_request_initial,
+)
+
 T = TypeVar('T')
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
 
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_data_versions_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_data_versions_operations.py
index 0505b56bf618..5e542b8824a8 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_data_versions_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_data_versions_operations.py
@@ -9,7 +9,13 @@
 from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar, Union
 
 from azure.core.async_paging import AsyncItemPaged, AsyncList
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import AsyncHttpResponse
 from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
@@ -21,7 +27,14 @@
 
 from ... import models as _models
 from ..._vendor import _convert_request
-from ...operations._data_versions_operations import build_create_or_update_request, build_delete_request, build_get_request, build_list_request, build_publish_request_initial
+from ...operations._data_versions_operations import (
+    build_create_or_update_request,
+    build_delete_request,
+    build_get_request,
+    build_list_request,
+    build_publish_request_initial,
+)
+
 T = TypeVar('T')
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
 
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_environment_versions_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_environment_versions_operations.py
index 9008f274f6ae..e8e7192df306 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_environment_versions_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_environment_versions_operations.py
@@ -9,7 +9,13 @@
 from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar, Union
 
 from azure.core.async_paging import AsyncItemPaged, AsyncList
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import AsyncHttpResponse
 from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
@@ -21,7 +27,14 @@
 
 from ... import models as _models
 from ..._vendor import _convert_request
-from ...operations._environment_versions_operations import build_create_or_update_request, build_delete_request, build_get_request, build_list_request, build_publish_request_initial
+from ...operations._environment_versions_operations import (
+    build_create_or_update_request,
+    build_delete_request,
+    build_get_request,
+    build_list_request,
+    build_publish_request_initial,
+)
+
 T = TypeVar('T')
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
 
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_jobs_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_jobs_operations.py
index 937f2b76f206..b79818aad95a 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_jobs_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_jobs_operations.py
@@ -9,7 +9,13 @@
 from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar, Union
 
 from azure.core.async_paging import AsyncItemPaged, AsyncList
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import AsyncHttpResponse
 from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
@@ -21,7 +27,15 @@
 
 from ... import models as _models
 from ..._vendor import _convert_request
-from ...operations._jobs_operations import build_cancel_request_initial, build_create_or_update_request, build_delete_request_initial, build_get_request, build_list_request, build_update_request
+from ...operations._jobs_operations import (
+    build_cancel_request_initial,
+    build_create_or_update_request,
+    build_delete_request_initial,
+    build_get_request,
+    build_list_request,
+    build_update_request,
+)
+
 T = TypeVar('T')
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
 
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_model_versions_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_model_versions_operations.py
index 9d084383ba16..eaa0d2b4a39b 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_model_versions_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_model_versions_operations.py
@@ -9,7 +9,13 @@
 from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar, Union
 
 from azure.core.async_paging import AsyncItemPaged, AsyncList
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import AsyncHttpResponse
 from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
@@ -21,7 +27,15 @@
 
 from ... import models as _models
 from ..._vendor import _convert_request
-from ...operations._model_versions_operations import build_create_or_update_request, build_delete_request, build_get_request, build_list_request, build_package_request_initial, build_publish_request_initial
+from ...operations._model_versions_operations import (
+    build_create_or_update_request,
+    build_delete_request,
+    build_get_request,
+    build_list_request,
+    build_package_request_initial,
+    build_publish_request_initial,
+)
+
 T = TypeVar('T')
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
 
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_registry_data_references_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_registry_data_references_operations.py
index a6180f2a946c..27cf5530b86f 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_registry_data_references_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_registry_data_references_operations.py
@@ -8,7 +8,13 @@
 # --------------------------------------------------------------------------
 from typing import Any, Callable, Dict, Optional, TypeVar
 
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import AsyncHttpResponse
 from azure.core.rest import HttpRequest
@@ -18,6 +24,7 @@
 from ... import models as _models
 from ..._vendor import _convert_request
 from ...operations._registry_data_references_operations import build_get_blob_reference_sas_request
+
 T = TypeVar('T')
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
 
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_schedules_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_schedules_operations.py
index 19585f1da233..a967ff77861a 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_schedules_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_schedules_operations.py
@@ -9,7 +9,13 @@
 from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar, Union
 
 from azure.core.async_paging import AsyncItemPaged, AsyncList
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import AsyncHttpResponse
 from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
@@ -21,7 +27,14 @@
 
 from ... import models as _models
 from ..._vendor import _convert_request
-from ...operations._schedules_operations import build_create_or_update_request_initial, build_delete_request_initial, build_get_request, build_list_request, build_trigger_request
+from ...operations._schedules_operations import (
+    build_create_or_update_request_initial,
+    build_delete_request_initial,
+    build_get_request,
+    build_list_request,
+    build_trigger_request,
+)
+
 T = TypeVar('T')
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
 
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_serverless_endpoints_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_serverless_endpoints_operations.py
index 0841846924d2..9901162d6e0b 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_serverless_endpoints_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_serverless_endpoints_operations.py
@@ -9,7 +9,13 @@
 from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar, Union
 
 from azure.core.async_paging import AsyncItemPaged, AsyncList
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import AsyncHttpResponse
 from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
@@ -21,7 +27,17 @@
 
 from ... import models as _models
 from ..._vendor import _convert_request
-from ...operations._serverless_endpoints_operations import build_create_or_update_request_initial, build_delete_request_initial, build_get_request, build_get_status_request, build_list_keys_request, build_list_request, build_regenerate_keys_request_initial, build_update_request_initial
+from ...operations._serverless_endpoints_operations import (
+    build_create_or_update_request_initial,
+    build_delete_request_initial,
+    build_get_request,
+    build_get_status_request,
+    build_list_keys_request,
+    build_list_request,
+    build_regenerate_keys_request_initial,
+    build_update_request_initial,
+)
+
 T = TypeVar('T')
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
 
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/__init__.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/__init__.py
index b5cc4237500b..35fb60cd6910 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/__init__.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/__init__.py
@@ -7,665 +7,667 @@
 # --------------------------------------------------------------------------
 
 try:
-    from ._models_py3 import AADAuthTypeWorkspaceConnectionProperties
-    from ._models_py3 import AKS
-    from ._models_py3 import AKSSchema
-    from ._models_py3 import AKSSchemaProperties
-    from ._models_py3 import AccessKeyAuthTypeWorkspaceConnectionProperties
-    from ._models_py3 import AccountApiKeys
-    from ._models_py3 import AccountKeyAuthTypeWorkspaceConnectionProperties
-    from ._models_py3 import AccountKeyDatastoreCredentials
-    from ._models_py3 import AccountKeyDatastoreSecrets
-    from ._models_py3 import AccountModel
-    from ._models_py3 import AcrDetails
-    from ._models_py3 import ActualCapacityInfo
-    from ._models_py3 import AksComputeSecrets
-    from ._models_py3 import AksComputeSecretsProperties
-    from ._models_py3 import AksNetworkingConfiguration
-    from ._models_py3 import AllFeatures
-    from ._models_py3 import AllNodes
-    from ._models_py3 import AmlCompute
-    from ._models_py3 import AmlComputeNodeInformation
-    from ._models_py3 import AmlComputeNodesInformation
-    from ._models_py3 import AmlComputeProperties
-    from ._models_py3 import AmlComputeSchema
-    from ._models_py3 import AmlToken
-    from ._models_py3 import AmlTokenComputeIdentity
-    from ._models_py3 import AmlUserFeature
-    from ._models_py3 import AnonymousAccessCredential
-    from ._models_py3 import ApiKeyAuthWorkspaceConnectionProperties
-    from ._models_py3 import ArmResourceId
-    from ._models_py3 import AssetBase
-    from ._models_py3 import AssetContainer
-    from ._models_py3 import AssetJobInput
-    from ._models_py3 import AssetJobOutput
-    from ._models_py3 import AssetReferenceBase
-    from ._models_py3 import AssignedUser
-    from ._models_py3 import AutoDeleteSetting
-    from ._models_py3 import AutoForecastHorizon
-    from ._models_py3 import AutoMLJob
-    from ._models_py3 import AutoMLVertical
-    from ._models_py3 import AutoNCrossValidations
-    from ._models_py3 import AutoPauseProperties
-    from ._models_py3 import AutoScaleProperties
-    from ._models_py3 import AutoSeasonality
-    from ._models_py3 import AutoTargetLags
-    from ._models_py3 import AutoTargetRollingWindowSize
-    from ._models_py3 import AutologgerSettings
-    from ._models_py3 import AzureBlobDatastore
-    from ._models_py3 import AzureDataLakeGen1Datastore
-    from ._models_py3 import AzureDataLakeGen2Datastore
-    from ._models_py3 import AzureDatastore
-    from ._models_py3 import AzureDevOpsWebhook
-    from ._models_py3 import AzureFileDatastore
-    from ._models_py3 import AzureMLBatchInferencingServer
-    from ._models_py3 import AzureMLOnlineInferencingServer
-    from ._models_py3 import AzureOpenAiFineTuning
-    from ._models_py3 import AzureOpenAiHyperParameters
-    from ._models_py3 import BanditPolicy
-    from ._models_py3 import BaseEnvironmentId
-    from ._models_py3 import BaseEnvironmentSource
-    from ._models_py3 import BatchDeployment
-    from ._models_py3 import BatchDeploymentConfiguration
-    from ._models_py3 import BatchDeploymentProperties
-    from ._models_py3 import BatchDeploymentTrackedResourceArmPaginatedResult
-    from ._models_py3 import BatchEndpoint
-    from ._models_py3 import BatchEndpointDefaults
-    from ._models_py3 import BatchEndpointProperties
-    from ._models_py3 import BatchEndpointTrackedResourceArmPaginatedResult
-    from ._models_py3 import BatchPipelineComponentDeploymentConfiguration
-    from ._models_py3 import BatchRetrySettings
-    from ._models_py3 import BayesianSamplingAlgorithm
-    from ._models_py3 import BindOptions
-    from ._models_py3 import BlobReferenceForConsumptionDto
-    from ._models_py3 import BuildContext
-    from ._models_py3 import CallRateLimit
-    from ._models_py3 import CapacityConfig
-    from ._models_py3 import CapacityReservationGroup
-    from ._models_py3 import CapacityReservationGroupProperties
-    from ._models_py3 import CapacityReservationGroupTrackedResourceArmPaginatedResult
-    from ._models_py3 import CategoricalDataDriftMetricThreshold
-    from ._models_py3 import CategoricalDataQualityMetricThreshold
-    from ._models_py3 import CategoricalPredictionDriftMetricThreshold
-    from ._models_py3 import CertificateDatastoreCredentials
-    from ._models_py3 import CertificateDatastoreSecrets
-    from ._models_py3 import Classification
-    from ._models_py3 import ClassificationModelPerformanceMetricThreshold
-    from ._models_py3 import ClassificationTrainingSettings
-    from ._models_py3 import ClusterUpdateParameters
-    from ._models_py3 import CocoExportSummary
-    from ._models_py3 import CodeConfiguration
-    from ._models_py3 import CodeContainer
-    from ._models_py3 import CodeContainerProperties
-    from ._models_py3 import CodeContainerResourceArmPaginatedResult
-    from ._models_py3 import CodeVersion
-    from ._models_py3 import CodeVersionProperties
-    from ._models_py3 import CodeVersionResourceArmPaginatedResult
-    from ._models_py3 import CognitiveServiceEndpointDeploymentResourceProperties
-    from ._models_py3 import CognitiveServicesSku
-    from ._models_py3 import Collection
-    from ._models_py3 import ColumnTransformer
-    from ._models_py3 import CommandJob
-    from ._models_py3 import CommandJobLimits
-    from ._models_py3 import ComponentConfiguration
-    from ._models_py3 import ComponentContainer
-    from ._models_py3 import ComponentContainerProperties
-    from ._models_py3 import ComponentContainerResourceArmPaginatedResult
-    from ._models_py3 import ComponentVersion
-    from ._models_py3 import ComponentVersionProperties
-    from ._models_py3 import ComponentVersionResourceArmPaginatedResult
-    from ._models_py3 import Compute
-    from ._models_py3 import ComputeInstance
-    from ._models_py3 import ComputeInstanceApplication
-    from ._models_py3 import ComputeInstanceAutologgerSettings
-    from ._models_py3 import ComputeInstanceConnectivityEndpoints
-    from ._models_py3 import ComputeInstanceContainer
-    from ._models_py3 import ComputeInstanceCreatedBy
-    from ._models_py3 import ComputeInstanceDataDisk
-    from ._models_py3 import ComputeInstanceDataMount
-    from ._models_py3 import ComputeInstanceEnvironmentInfo
-    from ._models_py3 import ComputeInstanceLastOperation
-    from ._models_py3 import ComputeInstanceProperties
-    from ._models_py3 import ComputeInstanceSchema
-    from ._models_py3 import ComputeInstanceSshSettings
-    from ._models_py3 import ComputeInstanceVersion
-    from ._models_py3 import ComputeRecurrenceSchedule
-    from ._models_py3 import ComputeResource
-    from ._models_py3 import ComputeResourceSchema
-    from ._models_py3 import ComputeRuntimeDto
-    from ._models_py3 import ComputeSchedules
-    from ._models_py3 import ComputeSecrets
-    from ._models_py3 import ComputeStartStopSchedule
-    from ._models_py3 import ContainerResourceRequirements
-    from ._models_py3 import ContainerResourceSettings
-    from ._models_py3 import ContentSafetyEndpointDeploymentResourceProperties
-    from ._models_py3 import ContentSafetyEndpointResourceProperties
-    from ._models_py3 import CosmosDbSettings
-    from ._models_py3 import CreateMonitorAction
-    from ._models_py3 import Cron
-    from ._models_py3 import CronTrigger
-    from ._models_py3 import CsvExportSummary
-    from ._models_py3 import CustomForecastHorizon
-    from ._models_py3 import CustomInferencingServer
-    from ._models_py3 import CustomKeys
-    from ._models_py3 import CustomKeysWorkspaceConnectionProperties
-    from ._models_py3 import CustomMetricThreshold
-    from ._models_py3 import CustomModelFineTuning
-    from ._models_py3 import CustomModelJobInput
-    from ._models_py3 import CustomModelJobOutput
-    from ._models_py3 import CustomMonitoringSignal
-    from ._models_py3 import CustomNCrossValidations
-    from ._models_py3 import CustomSeasonality
-    from ._models_py3 import CustomService
-    from ._models_py3 import CustomTargetLags
-    from ._models_py3 import CustomTargetRollingWindowSize
-    from ._models_py3 import DataCollector
-    from ._models_py3 import DataContainer
-    from ._models_py3 import DataContainerProperties
-    from ._models_py3 import DataContainerResourceArmPaginatedResult
-    from ._models_py3 import DataDriftMetricThresholdBase
-    from ._models_py3 import DataDriftMonitoringSignal
-    from ._models_py3 import DataFactory
-    from ._models_py3 import DataImport
-    from ._models_py3 import DataImportSource
-    from ._models_py3 import DataLakeAnalytics
-    from ._models_py3 import DataLakeAnalyticsSchema
-    from ._models_py3 import DataLakeAnalyticsSchemaProperties
-    from ._models_py3 import DataPathAssetReference
-    from ._models_py3 import DataQualityMetricThresholdBase
-    from ._models_py3 import DataQualityMonitoringSignal
-    from ._models_py3 import DataReferenceCredential
-    from ._models_py3 import DataVersionBase
-    from ._models_py3 import DataVersionBaseProperties
-    from ._models_py3 import DataVersionBaseResourceArmPaginatedResult
-    from ._models_py3 import DatabaseSource
-    from ._models_py3 import Databricks
-    from ._models_py3 import DatabricksComputeSecrets
-    from ._models_py3 import DatabricksComputeSecretsProperties
-    from ._models_py3 import DatabricksProperties
-    from ._models_py3 import DatabricksSchema
-    from ._models_py3 import DatasetExportSummary
-    from ._models_py3 import Datastore
-    from ._models_py3 import DatastoreCredentials
-    from ._models_py3 import DatastoreProperties
-    from ._models_py3 import DatastoreResourceArmPaginatedResult
-    from ._models_py3 import DatastoreSecrets
-    from ._models_py3 import DefaultScaleSettings
-    from ._models_py3 import DeploymentLogs
-    from ._models_py3 import DeploymentLogsRequest
-    from ._models_py3 import DeploymentModel
-    from ._models_py3 import DeploymentResourceConfiguration
-    from ._models_py3 import DestinationAsset
-    from ._models_py3 import DiagnoseRequestProperties
-    from ._models_py3 import DiagnoseResponseResult
-    from ._models_py3 import DiagnoseResponseResultValue
-    from ._models_py3 import DiagnoseResult
-    from ._models_py3 import DiagnoseWorkspaceParameters
-    from ._models_py3 import DistributionConfiguration
-    from ._models_py3 import Docker
-    from ._models_py3 import DockerCredential
-    from ._models_py3 import EarlyTerminationPolicy
-    from ._models_py3 import EncryptionKeyVaultUpdateProperties
-    from ._models_py3 import EncryptionProperty
-    from ._models_py3 import EncryptionUpdateProperties
-    from ._models_py3 import Endpoint
-    from ._models_py3 import EndpointAuthKeys
-    from ._models_py3 import EndpointAuthToken
-    from ._models_py3 import EndpointDeploymentModel
-    from ._models_py3 import EndpointDeploymentPropertiesBase
-    from ._models_py3 import EndpointDeploymentResourceProperties
-    from ._models_py3 import EndpointDeploymentResourcePropertiesBasicResource
-    from ._models_py3 import EndpointDeploymentResourcePropertiesBasicResourceArmPaginatedResult
-    from ._models_py3 import EndpointKeys
-    from ._models_py3 import EndpointModels
-    from ._models_py3 import EndpointPropertiesBase
-    from ._models_py3 import EndpointResourceProperties
-    from ._models_py3 import EndpointResourcePropertiesBasicResource
-    from ._models_py3 import EndpointResourcePropertiesBasicResourceArmPaginatedResult
-    from ._models_py3 import EndpointScheduleAction
-    from ._models_py3 import EnvironmentContainer
-    from ._models_py3 import EnvironmentContainerProperties
-    from ._models_py3 import EnvironmentContainerResourceArmPaginatedResult
-    from ._models_py3 import EnvironmentVariable
-    from ._models_py3 import EnvironmentVersion
-    from ._models_py3 import EnvironmentVersionProperties
-    from ._models_py3 import EnvironmentVersionResourceArmPaginatedResult
-    from ._models_py3 import ErrorAdditionalInfo
-    from ._models_py3 import ErrorDetail
-    from ._models_py3 import ErrorResponse
-    from ._models_py3 import EstimatedVMPrice
-    from ._models_py3 import EstimatedVMPrices
-    from ._models_py3 import ExportSummary
-    from ._models_py3 import ExternalFQDNResponse
-    from ._models_py3 import FQDNEndpoint
-    from ._models_py3 import FQDNEndpointDetail
-    from ._models_py3 import FQDNEndpoints
-    from ._models_py3 import FQDNEndpointsPropertyBag
-    from ._models_py3 import Feature
-    from ._models_py3 import FeatureAttributionDriftMonitoringSignal
-    from ._models_py3 import FeatureAttributionMetricThreshold
-    from ._models_py3 import FeatureImportanceSettings
-    from ._models_py3 import FeatureProperties
-    from ._models_py3 import FeatureResourceArmPaginatedResult
-    from ._models_py3 import FeatureStoreSettings
-    from ._models_py3 import FeatureSubset
-    from ._models_py3 import FeatureWindow
-    from ._models_py3 import FeaturesetContainer
-    from ._models_py3 import FeaturesetContainerProperties
-    from ._models_py3 import FeaturesetContainerResourceArmPaginatedResult
-    from ._models_py3 import FeaturesetSpecification
-    from ._models_py3 import FeaturesetVersion
-    from ._models_py3 import FeaturesetVersionBackfillRequest
-    from ._models_py3 import FeaturesetVersionBackfillResponse
-    from ._models_py3 import FeaturesetVersionProperties
-    from ._models_py3 import FeaturesetVersionResourceArmPaginatedResult
-    from ._models_py3 import FeaturestoreEntityContainer
-    from ._models_py3 import FeaturestoreEntityContainerProperties
-    from ._models_py3 import FeaturestoreEntityContainerResourceArmPaginatedResult
-    from ._models_py3 import FeaturestoreEntityVersion
-    from ._models_py3 import FeaturestoreEntityVersionProperties
-    from ._models_py3 import FeaturestoreEntityVersionResourceArmPaginatedResult
-    from ._models_py3 import FeaturizationSettings
-    from ._models_py3 import FileSystemSource
-    from ._models_py3 import FineTuningJob
-    from ._models_py3 import FineTuningVertical
-    from ._models_py3 import FixedInputData
-    from ._models_py3 import FlavorData
-    from ._models_py3 import ForecastHorizon
-    from ._models_py3 import Forecasting
-    from ._models_py3 import ForecastingSettings
-    from ._models_py3 import ForecastingTrainingSettings
-    from ._models_py3 import FqdnOutboundRule
-    from ._models_py3 import GenerationSafetyQualityMetricThreshold
-    from ._models_py3 import GenerationSafetyQualityMonitoringSignal
-    from ._models_py3 import GenerationTokenUsageMetricThreshold
-    from ._models_py3 import GenerationTokenUsageSignal
-    from ._models_py3 import GetBlobReferenceForConsumptionDto
-    from ._models_py3 import GetBlobReferenceSASRequestDto
-    from ._models_py3 import GetBlobReferenceSASResponseDto
-    from ._models_py3 import GridSamplingAlgorithm
-    from ._models_py3 import GroupStatus
-    from ._models_py3 import HDInsight
-    from ._models_py3 import HDInsightProperties
-    from ._models_py3 import HDInsightSchema
-    from ._models_py3 import HdfsDatastore
-    from ._models_py3 import IdAssetReference
-    from ._models_py3 import IdentityConfiguration
-    from ._models_py3 import IdentityForCmk
-    from ._models_py3 import IdleShutdownSetting
-    from ._models_py3 import Image
-    from ._models_py3 import ImageClassification
-    from ._models_py3 import ImageClassificationBase
-    from ._models_py3 import ImageClassificationMultilabel
-    from ._models_py3 import ImageInstanceSegmentation
-    from ._models_py3 import ImageLimitSettings
-    from ._models_py3 import ImageMetadata
-    from ._models_py3 import ImageModelDistributionSettings
-    from ._models_py3 import ImageModelDistributionSettingsClassification
-    from ._models_py3 import ImageModelDistributionSettingsObjectDetection
-    from ._models_py3 import ImageModelSettings
-    from ._models_py3 import ImageModelSettingsClassification
-    from ._models_py3 import ImageModelSettingsObjectDetection
-    from ._models_py3 import ImageObjectDetection
-    from ._models_py3 import ImageObjectDetectionBase
-    from ._models_py3 import ImageSweepSettings
-    from ._models_py3 import ImageVertical
-    from ._models_py3 import ImportDataAction
-    from ._models_py3 import IndexColumn
-    from ._models_py3 import InferenceContainerProperties
-    from ._models_py3 import InferenceEndpoint
-    from ._models_py3 import InferenceEndpointProperties
-    from ._models_py3 import InferenceEndpointTrackedResourceArmPaginatedResult
-    from ._models_py3 import InferenceGroup
-    from ._models_py3 import InferenceGroupProperties
-    from ._models_py3 import InferenceGroupTrackedResourceArmPaginatedResult
-    from ._models_py3 import InferencePool
-    from ._models_py3 import InferencePoolProperties
-    from ._models_py3 import InferencePoolTrackedResourceArmPaginatedResult
-    from ._models_py3 import InferencingServer
-    from ._models_py3 import InstanceTypeSchema
-    from ._models_py3 import InstanceTypeSchemaResources
-    from ._models_py3 import IntellectualProperty
-    from ._models_py3 import JobBase
-    from ._models_py3 import JobBaseProperties
-    from ._models_py3 import JobBaseResourceArmPaginatedResult
-    from ._models_py3 import JobInput
-    from ._models_py3 import JobLimits
-    from ._models_py3 import JobOutput
-    from ._models_py3 import JobResourceConfiguration
-    from ._models_py3 import JobScheduleAction
-    from ._models_py3 import JobService
-    from ._models_py3 import JupyterKernelConfig
-    from ._models_py3 import KerberosCredentials
-    from ._models_py3 import KerberosKeytabCredentials
-    from ._models_py3 import KerberosKeytabSecrets
-    from ._models_py3 import KerberosPasswordCredentials
-    from ._models_py3 import KerberosPasswordSecrets
-    from ._models_py3 import KeyVaultProperties
-    from ._models_py3 import Kubernetes
-    from ._models_py3 import KubernetesOnlineDeployment
-    from ._models_py3 import KubernetesProperties
-    from ._models_py3 import KubernetesSchema
-    from ._models_py3 import LabelCategory
-    from ._models_py3 import LabelClass
-    from ._models_py3 import LabelingDataConfiguration
-    from ._models_py3 import LabelingJob
-    from ._models_py3 import LabelingJobImageProperties
-    from ._models_py3 import LabelingJobInstructions
-    from ._models_py3 import LabelingJobMediaProperties
-    from ._models_py3 import LabelingJobProperties
-    from ._models_py3 import LabelingJobResourceArmPaginatedResult
-    from ._models_py3 import LabelingJobTextProperties
-    from ._models_py3 import LakeHouseArtifact
-    from ._models_py3 import ListAmlUserFeatureResult
-    from ._models_py3 import ListNotebookKeysResult
-    from ._models_py3 import ListStorageAccountKeysResult
-    from ._models_py3 import ListUsagesResult
-    from ._models_py3 import ListWorkspaceKeysResult
-    from ._models_py3 import ListWorkspaceQuotas
-    from ._models_py3 import LiteralJobInput
-    from ._models_py3 import MLAssistConfiguration
-    from ._models_py3 import MLAssistConfigurationDisabled
-    from ._models_py3 import MLAssistConfigurationEnabled
-    from ._models_py3 import MLFlowModelJobInput
-    from ._models_py3 import MLFlowModelJobOutput
-    from ._models_py3 import MLTableData
-    from ._models_py3 import MLTableJobInput
-    from ._models_py3 import MLTableJobOutput
-    from ._models_py3 import ManagedComputeIdentity
-    from ._models_py3 import ManagedIdentity
-    from ._models_py3 import ManagedIdentityAuthTypeWorkspaceConnectionProperties
-    from ._models_py3 import ManagedIdentityCredential
-    from ._models_py3 import ManagedNetworkProvisionOptions
-    from ._models_py3 import ManagedNetworkProvisionStatus
-    from ._models_py3 import ManagedNetworkSettings
-    from ._models_py3 import ManagedOnlineDeployment
-    from ._models_py3 import ManagedOnlineEndpointDeploymentResourceProperties
-    from ._models_py3 import ManagedOnlineEndpointResourceProperties
-    from ._models_py3 import ManagedResourceGroupAssignedIdentities
-    from ._models_py3 import ManagedResourceGroupSettings
-    from ._models_py3 import ManagedServiceIdentity
-    from ._models_py3 import MarketplacePlan
-    from ._models_py3 import MarketplaceSubscription
-    from ._models_py3 import MarketplaceSubscriptionProperties
-    from ._models_py3 import MarketplaceSubscriptionResourceArmPaginatedResult
-    from ._models_py3 import MaterializationComputeResource
-    from ._models_py3 import MaterializationSettings
-    from ._models_py3 import MedianStoppingPolicy
-    from ._models_py3 import ModelConfiguration
-    from ._models_py3 import ModelContainer
-    from ._models_py3 import ModelContainerProperties
-    from ._models_py3 import ModelContainerResourceArmPaginatedResult
-    from ._models_py3 import ModelDeprecationInfo
-    from ._models_py3 import ModelPackageInput
-    from ._models_py3 import ModelPerformanceMetricThresholdBase
-    from ._models_py3 import ModelPerformanceSignal
-    from ._models_py3 import ModelSettings
-    from ._models_py3 import ModelSku
-    from ._models_py3 import ModelVersion
-    from ._models_py3 import ModelVersionProperties
-    from ._models_py3 import ModelVersionResourceArmPaginatedResult
-    from ._models_py3 import MonitorComputeConfigurationBase
-    from ._models_py3 import MonitorComputeIdentityBase
-    from ._models_py3 import MonitorDefinition
-    from ._models_py3 import MonitorEmailNotificationSettings
-    from ._models_py3 import MonitorNotificationSettings
-    from ._models_py3 import MonitorServerlessSparkCompute
-    from ._models_py3 import MonitoringDataSegment
-    from ._models_py3 import MonitoringFeatureFilterBase
-    from ._models_py3 import MonitoringInputDataBase
-    from ._models_py3 import MonitoringSignalBase
-    from ._models_py3 import MonitoringTarget
-    from ._models_py3 import MonitoringThreshold
-    from ._models_py3 import MonitoringWorkspaceConnection
-    from ._models_py3 import Mpi
-    from ._models_py3 import NCrossValidations
-    from ._models_py3 import NlpFixedParameters
-    from ._models_py3 import NlpParameterSubspace
-    from ._models_py3 import NlpSweepSettings
-    from ._models_py3 import NlpVertical
-    from ._models_py3 import NlpVerticalFeaturizationSettings
-    from ._models_py3 import NlpVerticalLimitSettings
-    from ._models_py3 import NodeStateCounts
-    from ._models_py3 import Nodes
-    from ._models_py3 import NoneAuthTypeWorkspaceConnectionProperties
-    from ._models_py3 import NoneDatastoreCredentials
-    from ._models_py3 import NotebookAccessTokenResult
-    from ._models_py3 import NotebookPreparationError
-    from ._models_py3 import NotebookResourceInfo
-    from ._models_py3 import NotificationSetting
-    from ._models_py3 import NumericalDataDriftMetricThreshold
-    from ._models_py3 import NumericalDataQualityMetricThreshold
-    from ._models_py3 import NumericalPredictionDriftMetricThreshold
-    from ._models_py3 import OAuth2AuthTypeWorkspaceConnectionProperties
-    from ._models_py3 import Objective
-    from ._models_py3 import OneLakeArtifact
-    from ._models_py3 import OneLakeDatastore
-    from ._models_py3 import OnlineDeployment
-    from ._models_py3 import OnlineDeploymentProperties
-    from ._models_py3 import OnlineDeploymentTrackedResourceArmPaginatedResult
-    from ._models_py3 import OnlineEndpoint
-    from ._models_py3 import OnlineEndpointProperties
-    from ._models_py3 import OnlineEndpointTrackedResourceArmPaginatedResult
-    from ._models_py3 import OnlineInferenceConfiguration
-    from ._models_py3 import OnlineRequestSettings
-    from ._models_py3 import OnlineScaleSettings
-    from ._models_py3 import OpenAIEndpointDeploymentResourceProperties
-    from ._models_py3 import OpenAIEndpointResourceProperties
-    from ._models_py3 import Operation
-    from ._models_py3 import OperationDisplay
-    from ._models_py3 import OperationListResult
-    from ._models_py3 import OsPatchingStatus
-    from ._models_py3 import OutboundRule
-    from ._models_py3 import OutboundRuleBasicResource
-    from ._models_py3 import OutboundRuleListResult
-    from ._models_py3 import OutputPathAssetReference
-    from ._models_py3 import PATAuthTypeWorkspaceConnectionProperties
-    from ._models_py3 import PackageInputPathBase
-    from ._models_py3 import PackageInputPathId
-    from ._models_py3 import PackageInputPathUrl
-    from ._models_py3 import PackageInputPathVersion
-    from ._models_py3 import PackageRequest
-    from ._models_py3 import PackageResponse
-    from ._models_py3 import PaginatedComputeResourcesList
-    from ._models_py3 import PartialBatchDeployment
-    from ._models_py3 import PartialBatchDeploymentPartialMinimalTrackedResourceWithProperties
-    from ._models_py3 import PartialJobBase
-    from ._models_py3 import PartialJobBasePartialResource
-    from ._models_py3 import PartialManagedServiceIdentity
-    from ._models_py3 import PartialMinimalTrackedResource
-    from ._models_py3 import PartialMinimalTrackedResourceWithIdentity
-    from ._models_py3 import PartialMinimalTrackedResourceWithSku
-    from ._models_py3 import PartialMinimalTrackedResourceWithSkuAndIdentity
-    from ._models_py3 import PartialNotificationSetting
-    from ._models_py3 import PartialRegistryPartialTrackedResource
-    from ._models_py3 import PartialSku
-    from ._models_py3 import Password
-    from ._models_py3 import PendingUploadCredentialDto
-    from ._models_py3 import PendingUploadRequestDto
-    from ._models_py3 import PendingUploadResponseDto
-    from ._models_py3 import PersonalComputeInstanceSettings
-    from ._models_py3 import PipelineJob
-    from ._models_py3 import PoolEnvironmentConfiguration
-    from ._models_py3 import PoolModelConfiguration
-    from ._models_py3 import PoolStatus
-    from ._models_py3 import PredictionDriftMetricThresholdBase
-    from ._models_py3 import PredictionDriftMonitoringSignal
-    from ._models_py3 import PrivateEndpoint
-    from ._models_py3 import PrivateEndpointConnection
-    from ._models_py3 import PrivateEndpointConnectionListResult
-    from ._models_py3 import PrivateEndpointDestination
-    from ._models_py3 import PrivateEndpointOutboundRule
-    from ._models_py3 import PrivateEndpointResource
-    from ._models_py3 import PrivateLinkResource
-    from ._models_py3 import PrivateLinkResourceListResult
-    from ._models_py3 import PrivateLinkServiceConnectionState
-    from ._models_py3 import ProbeSettings
-    from ._models_py3 import ProgressMetrics
-    from ._models_py3 import PropertiesBase
-    from ._models_py3 import ProxyResource
-    from ._models_py3 import PyTorch
-    from ._models_py3 import QueueSettings
-    from ._models_py3 import QuotaBaseProperties
-    from ._models_py3 import QuotaUpdateParameters
-    from ._models_py3 import RandomSamplingAlgorithm
-    from ._models_py3 import Ray
-    from ._models_py3 import Recurrence
-    from ._models_py3 import RecurrenceSchedule
-    from ._models_py3 import RecurrenceTrigger
-    from ._models_py3 import RegenerateEndpointKeysRequest
-    from ._models_py3 import RegenerateServiceAccountKeyContent
-    from ._models_py3 import Registry
-    from ._models_py3 import RegistryListCredentialsResult
-    from ._models_py3 import RegistryPartialManagedServiceIdentity
-    from ._models_py3 import RegistryPrivateEndpointConnection
-    from ._models_py3 import RegistryPrivateLinkServiceConnectionState
-    from ._models_py3 import RegistryRegionArmDetails
-    from ._models_py3 import RegistryTrackedResourceArmPaginatedResult
-    from ._models_py3 import Regression
-    from ._models_py3 import RegressionModelPerformanceMetricThreshold
-    from ._models_py3 import RegressionTrainingSettings
-    from ._models_py3 import RequestConfiguration
-    from ._models_py3 import RequestLogging
-    from ._models_py3 import RequestMatchPattern
-    from ._models_py3 import ResizeSchema
-    from ._models_py3 import Resource
-    from ._models_py3 import ResourceBase
-    from ._models_py3 import ResourceConfiguration
-    from ._models_py3 import ResourceId
-    from ._models_py3 import ResourceName
-    from ._models_py3 import ResourceQuota
-    from ._models_py3 import RollingInputData
-    from ._models_py3 import Route
-    from ._models_py3 import SASAuthTypeWorkspaceConnectionProperties
-    from ._models_py3 import SASCredential
-    from ._models_py3 import SASCredentialDto
-    from ._models_py3 import SamplingAlgorithm
-    from ._models_py3 import SasDatastoreCredentials
-    from ._models_py3 import SasDatastoreSecrets
-    from ._models_py3 import ScaleSettings
-    from ._models_py3 import ScaleSettingsInformation
-    from ._models_py3 import Schedule
-    from ._models_py3 import ScheduleActionBase
-    from ._models_py3 import ScheduleBase
-    from ._models_py3 import ScheduleProperties
-    from ._models_py3 import ScheduleResourceArmPaginatedResult
-    from ._models_py3 import ScriptReference
-    from ._models_py3 import ScriptsToExecute
-    from ._models_py3 import Seasonality
-    from ._models_py3 import SecretConfiguration
-    from ._models_py3 import ServerlessComputeSettings
-    from ._models_py3 import ServerlessEndpoint
-    from ._models_py3 import ServerlessEndpointCapacityReservation
-    from ._models_py3 import ServerlessEndpointProperties
-    from ._models_py3 import ServerlessEndpointStatus
-    from ._models_py3 import ServerlessEndpointTrackedResourceArmPaginatedResult
-    from ._models_py3 import ServerlessInferenceEndpoint
-    from ._models_py3 import ServerlessOffer
-    from ._models_py3 import ServiceManagedResourcesSettings
-    from ._models_py3 import ServicePrincipalAuthTypeWorkspaceConnectionProperties
-    from ._models_py3 import ServicePrincipalDatastoreCredentials
-    from ._models_py3 import ServicePrincipalDatastoreSecrets
-    from ._models_py3 import ServiceTagDestination
-    from ._models_py3 import ServiceTagOutboundRule
-    from ._models_py3 import SetupScripts
-    from ._models_py3 import SharedPrivateLinkResource
-    from ._models_py3 import Sku
-    from ._models_py3 import SkuCapacity
-    from ._models_py3 import SkuResource
-    from ._models_py3 import SkuResourceArmPaginatedResult
-    from ._models_py3 import SkuSetting
-    from ._models_py3 import SparkJob
-    from ._models_py3 import SparkJobEntry
-    from ._models_py3 import SparkJobPythonEntry
-    from ._models_py3 import SparkJobScalaEntry
-    from ._models_py3 import SparkResourceConfiguration
-    from ._models_py3 import SpeechEndpointDeploymentResourceProperties
-    from ._models_py3 import SpeechEndpointResourceProperties
-    from ._models_py3 import SslConfiguration
-    from ._models_py3 import StackEnsembleSettings
-    from ._models_py3 import StaticInputData
-    from ._models_py3 import StatusMessage
-    from ._models_py3 import StorageAccountDetails
-    from ._models_py3 import SweepJob
-    from ._models_py3 import SweepJobLimits
-    from ._models_py3 import SynapseSpark
-    from ._models_py3 import SynapseSparkProperties
-    from ._models_py3 import SystemCreatedAcrAccount
-    from ._models_py3 import SystemCreatedStorageAccount
-    from ._models_py3 import SystemData
-    from ._models_py3 import SystemService
-    from ._models_py3 import TableFixedParameters
-    from ._models_py3 import TableParameterSubspace
-    from ._models_py3 import TableSweepSettings
-    from ._models_py3 import TableVertical
-    from ._models_py3 import TableVerticalFeaturizationSettings
-    from ._models_py3 import TableVerticalLimitSettings
-    from ._models_py3 import TargetLags
-    from ._models_py3 import TargetRollingWindowSize
-    from ._models_py3 import TargetUtilizationScaleSettings
-    from ._models_py3 import TensorFlow
-    from ._models_py3 import TextClassification
-    from ._models_py3 import TextClassificationMultilabel
-    from ._models_py3 import TextNer
-    from ._models_py3 import ThrottlingRule
-    from ._models_py3 import TmpfsOptions
-    from ._models_py3 import TopNFeaturesByAttribution
-    from ._models_py3 import TrackedResource
-    from ._models_py3 import TrainingSettings
-    from ._models_py3 import TrialComponent
-    from ._models_py3 import TriggerBase
-    from ._models_py3 import TriggerOnceRequest
-    from ._models_py3 import TriggerRunSubmissionDto
-    from ._models_py3 import TritonInferencingServer
-    from ._models_py3 import TritonModelJobInput
-    from ._models_py3 import TritonModelJobOutput
-    from ._models_py3 import TruncationSelectionPolicy
-    from ._models_py3 import UpdateWorkspaceQuotas
-    from ._models_py3 import UpdateWorkspaceQuotasResult
-    from ._models_py3 import UriFileDataVersion
-    from ._models_py3 import UriFileJobInput
-    from ._models_py3 import UriFileJobOutput
-    from ._models_py3 import UriFolderDataVersion
-    from ._models_py3 import UriFolderJobInput
-    from ._models_py3 import UriFolderJobOutput
-    from ._models_py3 import Usage
-    from ._models_py3 import UsageName
-    from ._models_py3 import UserAccountCredentials
-    from ._models_py3 import UserAssignedIdentity
-    from ._models_py3 import UserCreatedAcrAccount
-    from ._models_py3 import UserCreatedStorageAccount
-    from ._models_py3 import UserIdentity
-    from ._models_py3 import UsernamePasswordAuthTypeWorkspaceConnectionProperties
-    from ._models_py3 import VirtualMachine
-    from ._models_py3 import VirtualMachineImage
-    from ._models_py3 import VirtualMachineSchema
-    from ._models_py3 import VirtualMachineSchemaProperties
-    from ._models_py3 import VirtualMachineSecrets
-    from ._models_py3 import VirtualMachineSecretsSchema
-    from ._models_py3 import VirtualMachineSize
-    from ._models_py3 import VirtualMachineSizeListResult
-    from ._models_py3 import VirtualMachineSshCredentials
-    from ._models_py3 import VolumeDefinition
-    from ._models_py3 import VolumeOptions
-    from ._models_py3 import Webhook
-    from ._models_py3 import Workspace
-    from ._models_py3 import WorkspaceConnectionAccessKey
-    from ._models_py3 import WorkspaceConnectionApiKey
-    from ._models_py3 import WorkspaceConnectionManagedIdentity
-    from ._models_py3 import WorkspaceConnectionOAuth2
-    from ._models_py3 import WorkspaceConnectionPersonalAccessToken
-    from ._models_py3 import WorkspaceConnectionPropertiesV2
-    from ._models_py3 import WorkspaceConnectionPropertiesV2BasicResource
-    from ._models_py3 import WorkspaceConnectionPropertiesV2BasicResourceArmPaginatedResult
-    from ._models_py3 import WorkspaceConnectionServicePrincipal
-    from ._models_py3 import WorkspaceConnectionSharedAccessSignature
-    from ._models_py3 import WorkspaceConnectionUpdateParameter
-    from ._models_py3 import WorkspaceConnectionUsernamePassword
-    from ._models_py3 import WorkspaceHubConfig
-    from ._models_py3 import WorkspaceListResult
-    from ._models_py3 import WorkspacePrivateEndpointResource
-    from ._models_py3 import WorkspaceUpdateParameters
+    from ._models_py3 import (
+        AKS,
+        AADAuthTypeWorkspaceConnectionProperties,
+        AccessKeyAuthTypeWorkspaceConnectionProperties,
+        AccountApiKeys,
+        AccountKeyAuthTypeWorkspaceConnectionProperties,
+        AccountKeyDatastoreCredentials,
+        AccountKeyDatastoreSecrets,
+        AccountModel,
+        AcrDetails,
+        ActualCapacityInfo,
+        AksComputeSecrets,
+        AksComputeSecretsProperties,
+        AksNetworkingConfiguration,
+        AKSSchema,
+        AKSSchemaProperties,
+        AllFeatures,
+        AllNodes,
+        AmlCompute,
+        AmlComputeNodeInformation,
+        AmlComputeNodesInformation,
+        AmlComputeProperties,
+        AmlComputeSchema,
+        AmlToken,
+        AmlTokenComputeIdentity,
+        AmlUserFeature,
+        AnonymousAccessCredential,
+        ApiKeyAuthWorkspaceConnectionProperties,
+        ArmResourceId,
+        AssetBase,
+        AssetContainer,
+        AssetJobInput,
+        AssetJobOutput,
+        AssetReferenceBase,
+        AssignedUser,
+        AutoDeleteSetting,
+        AutoForecastHorizon,
+        AutologgerSettings,
+        AutoMLJob,
+        AutoMLVertical,
+        AutoNCrossValidations,
+        AutoPauseProperties,
+        AutoScaleProperties,
+        AutoSeasonality,
+        AutoTargetLags,
+        AutoTargetRollingWindowSize,
+        AzureBlobDatastore,
+        AzureDataLakeGen1Datastore,
+        AzureDataLakeGen2Datastore,
+        AzureDatastore,
+        AzureDevOpsWebhook,
+        AzureFileDatastore,
+        AzureMLBatchInferencingServer,
+        AzureMLOnlineInferencingServer,
+        AzureOpenAiFineTuning,
+        AzureOpenAiHyperParameters,
+        BanditPolicy,
+        BaseEnvironmentId,
+        BaseEnvironmentSource,
+        BatchDeployment,
+        BatchDeploymentConfiguration,
+        BatchDeploymentProperties,
+        BatchDeploymentTrackedResourceArmPaginatedResult,
+        BatchEndpoint,
+        BatchEndpointDefaults,
+        BatchEndpointProperties,
+        BatchEndpointTrackedResourceArmPaginatedResult,
+        BatchPipelineComponentDeploymentConfiguration,
+        BatchRetrySettings,
+        BayesianSamplingAlgorithm,
+        BindOptions,
+        BlobReferenceForConsumptionDto,
+        BuildContext,
+        CallRateLimit,
+        CapacityConfig,
+        CapacityReservationGroup,
+        CapacityReservationGroupProperties,
+        CapacityReservationGroupTrackedResourceArmPaginatedResult,
+        CategoricalDataDriftMetricThreshold,
+        CategoricalDataQualityMetricThreshold,
+        CategoricalPredictionDriftMetricThreshold,
+        CertificateDatastoreCredentials,
+        CertificateDatastoreSecrets,
+        Classification,
+        ClassificationModelPerformanceMetricThreshold,
+        ClassificationTrainingSettings,
+        ClusterUpdateParameters,
+        CocoExportSummary,
+        CodeConfiguration,
+        CodeContainer,
+        CodeContainerProperties,
+        CodeContainerResourceArmPaginatedResult,
+        CodeVersion,
+        CodeVersionProperties,
+        CodeVersionResourceArmPaginatedResult,
+        CognitiveServiceEndpointDeploymentResourceProperties,
+        CognitiveServicesSku,
+        Collection,
+        ColumnTransformer,
+        CommandJob,
+        CommandJobLimits,
+        ComponentConfiguration,
+        ComponentContainer,
+        ComponentContainerProperties,
+        ComponentContainerResourceArmPaginatedResult,
+        ComponentVersion,
+        ComponentVersionProperties,
+        ComponentVersionResourceArmPaginatedResult,
+        Compute,
+        ComputeInstance,
+        ComputeInstanceApplication,
+        ComputeInstanceAutologgerSettings,
+        ComputeInstanceConnectivityEndpoints,
+        ComputeInstanceContainer,
+        ComputeInstanceCreatedBy,
+        ComputeInstanceDataDisk,
+        ComputeInstanceDataMount,
+        ComputeInstanceEnvironmentInfo,
+        ComputeInstanceLastOperation,
+        ComputeInstanceProperties,
+        ComputeInstanceSchema,
+        ComputeInstanceSshSettings,
+        ComputeInstanceVersion,
+        ComputeRecurrenceSchedule,
+        ComputeResource,
+        ComputeResourceSchema,
+        ComputeRuntimeDto,
+        ComputeSchedules,
+        ComputeSecrets,
+        ComputeStartStopSchedule,
+        ContainerResourceRequirements,
+        ContainerResourceSettings,
+        ContentSafetyEndpointDeploymentResourceProperties,
+        ContentSafetyEndpointResourceProperties,
+        CosmosDbSettings,
+        CreateMonitorAction,
+        Cron,
+        CronTrigger,
+        CsvExportSummary,
+        CustomForecastHorizon,
+        CustomInferencingServer,
+        CustomKeys,
+        CustomKeysWorkspaceConnectionProperties,
+        CustomMetricThreshold,
+        CustomModelFineTuning,
+        CustomModelJobInput,
+        CustomModelJobOutput,
+        CustomMonitoringSignal,
+        CustomNCrossValidations,
+        CustomSeasonality,
+        CustomService,
+        CustomTargetLags,
+        CustomTargetRollingWindowSize,
+        DatabaseSource,
+        Databricks,
+        DatabricksComputeSecrets,
+        DatabricksComputeSecretsProperties,
+        DatabricksProperties,
+        DatabricksSchema,
+        DataCollector,
+        DataContainer,
+        DataContainerProperties,
+        DataContainerResourceArmPaginatedResult,
+        DataDriftMetricThresholdBase,
+        DataDriftMonitoringSignal,
+        DataFactory,
+        DataImport,
+        DataImportSource,
+        DataLakeAnalytics,
+        DataLakeAnalyticsSchema,
+        DataLakeAnalyticsSchemaProperties,
+        DataPathAssetReference,
+        DataQualityMetricThresholdBase,
+        DataQualityMonitoringSignal,
+        DataReferenceCredential,
+        DatasetExportSummary,
+        Datastore,
+        DatastoreCredentials,
+        DatastoreProperties,
+        DatastoreResourceArmPaginatedResult,
+        DatastoreSecrets,
+        DataVersionBase,
+        DataVersionBaseProperties,
+        DataVersionBaseResourceArmPaginatedResult,
+        DefaultScaleSettings,
+        DeploymentLogs,
+        DeploymentLogsRequest,
+        DeploymentModel,
+        DeploymentResourceConfiguration,
+        DestinationAsset,
+        DiagnoseRequestProperties,
+        DiagnoseResponseResult,
+        DiagnoseResponseResultValue,
+        DiagnoseResult,
+        DiagnoseWorkspaceParameters,
+        DistributionConfiguration,
+        Docker,
+        DockerCredential,
+        EarlyTerminationPolicy,
+        EncryptionKeyVaultUpdateProperties,
+        EncryptionProperty,
+        EncryptionUpdateProperties,
+        Endpoint,
+        EndpointAuthKeys,
+        EndpointAuthToken,
+        EndpointDeploymentModel,
+        EndpointDeploymentPropertiesBase,
+        EndpointDeploymentResourceProperties,
+        EndpointDeploymentResourcePropertiesBasicResource,
+        EndpointDeploymentResourcePropertiesBasicResourceArmPaginatedResult,
+        EndpointKeys,
+        EndpointModels,
+        EndpointPropertiesBase,
+        EndpointResourceProperties,
+        EndpointResourcePropertiesBasicResource,
+        EndpointResourcePropertiesBasicResourceArmPaginatedResult,
+        EndpointScheduleAction,
+        EnvironmentContainer,
+        EnvironmentContainerProperties,
+        EnvironmentContainerResourceArmPaginatedResult,
+        EnvironmentVariable,
+        EnvironmentVersion,
+        EnvironmentVersionProperties,
+        EnvironmentVersionResourceArmPaginatedResult,
+        ErrorAdditionalInfo,
+        ErrorDetail,
+        ErrorResponse,
+        EstimatedVMPrice,
+        EstimatedVMPrices,
+        ExportSummary,
+        ExternalFQDNResponse,
+        Feature,
+        FeatureAttributionDriftMonitoringSignal,
+        FeatureAttributionMetricThreshold,
+        FeatureImportanceSettings,
+        FeatureProperties,
+        FeatureResourceArmPaginatedResult,
+        FeaturesetContainer,
+        FeaturesetContainerProperties,
+        FeaturesetContainerResourceArmPaginatedResult,
+        FeaturesetSpecification,
+        FeaturesetVersion,
+        FeaturesetVersionBackfillRequest,
+        FeaturesetVersionBackfillResponse,
+        FeaturesetVersionProperties,
+        FeaturesetVersionResourceArmPaginatedResult,
+        FeaturestoreEntityContainer,
+        FeaturestoreEntityContainerProperties,
+        FeaturestoreEntityContainerResourceArmPaginatedResult,
+        FeaturestoreEntityVersion,
+        FeaturestoreEntityVersionProperties,
+        FeaturestoreEntityVersionResourceArmPaginatedResult,
+        FeatureStoreSettings,
+        FeatureSubset,
+        FeatureWindow,
+        FeaturizationSettings,
+        FileSystemSource,
+        FineTuningJob,
+        FineTuningVertical,
+        FixedInputData,
+        FlavorData,
+        ForecastHorizon,
+        Forecasting,
+        ForecastingSettings,
+        ForecastingTrainingSettings,
+        FQDNEndpoint,
+        FQDNEndpointDetail,
+        FQDNEndpoints,
+        FQDNEndpointsPropertyBag,
+        FqdnOutboundRule,
+        GenerationSafetyQualityMetricThreshold,
+        GenerationSafetyQualityMonitoringSignal,
+        GenerationTokenUsageMetricThreshold,
+        GenerationTokenUsageSignal,
+        GetBlobReferenceForConsumptionDto,
+        GetBlobReferenceSASRequestDto,
+        GetBlobReferenceSASResponseDto,
+        GridSamplingAlgorithm,
+        GroupStatus,
+        HdfsDatastore,
+        HDInsight,
+        HDInsightProperties,
+        HDInsightSchema,
+        IdAssetReference,
+        IdentityConfiguration,
+        IdentityForCmk,
+        IdleShutdownSetting,
+        Image,
+        ImageClassification,
+        ImageClassificationBase,
+        ImageClassificationMultilabel,
+        ImageInstanceSegmentation,
+        ImageLimitSettings,
+        ImageMetadata,
+        ImageModelDistributionSettings,
+        ImageModelDistributionSettingsClassification,
+        ImageModelDistributionSettingsObjectDetection,
+        ImageModelSettings,
+        ImageModelSettingsClassification,
+        ImageModelSettingsObjectDetection,
+        ImageObjectDetection,
+        ImageObjectDetectionBase,
+        ImageSweepSettings,
+        ImageVertical,
+        ImportDataAction,
+        IndexColumn,
+        InferenceContainerProperties,
+        InferenceEndpoint,
+        InferenceEndpointProperties,
+        InferenceEndpointTrackedResourceArmPaginatedResult,
+        InferenceGroup,
+        InferenceGroupProperties,
+        InferenceGroupTrackedResourceArmPaginatedResult,
+        InferencePool,
+        InferencePoolProperties,
+        InferencePoolTrackedResourceArmPaginatedResult,
+        InferencingServer,
+        InstanceTypeSchema,
+        InstanceTypeSchemaResources,
+        IntellectualProperty,
+        JobBase,
+        JobBaseProperties,
+        JobBaseResourceArmPaginatedResult,
+        JobInput,
+        JobLimits,
+        JobOutput,
+        JobResourceConfiguration,
+        JobScheduleAction,
+        JobService,
+        JupyterKernelConfig,
+        KerberosCredentials,
+        KerberosKeytabCredentials,
+        KerberosKeytabSecrets,
+        KerberosPasswordCredentials,
+        KerberosPasswordSecrets,
+        KeyVaultProperties,
+        Kubernetes,
+        KubernetesOnlineDeployment,
+        KubernetesProperties,
+        KubernetesSchema,
+        LabelCategory,
+        LabelClass,
+        LabelingDataConfiguration,
+        LabelingJob,
+        LabelingJobImageProperties,
+        LabelingJobInstructions,
+        LabelingJobMediaProperties,
+        LabelingJobProperties,
+        LabelingJobResourceArmPaginatedResult,
+        LabelingJobTextProperties,
+        LakeHouseArtifact,
+        ListAmlUserFeatureResult,
+        ListNotebookKeysResult,
+        ListStorageAccountKeysResult,
+        ListUsagesResult,
+        ListWorkspaceKeysResult,
+        ListWorkspaceQuotas,
+        LiteralJobInput,
+        ManagedComputeIdentity,
+        ManagedIdentity,
+        ManagedIdentityAuthTypeWorkspaceConnectionProperties,
+        ManagedIdentityCredential,
+        ManagedNetworkProvisionOptions,
+        ManagedNetworkProvisionStatus,
+        ManagedNetworkSettings,
+        ManagedOnlineDeployment,
+        ManagedOnlineEndpointDeploymentResourceProperties,
+        ManagedOnlineEndpointResourceProperties,
+        ManagedResourceGroupAssignedIdentities,
+        ManagedResourceGroupSettings,
+        ManagedServiceIdentity,
+        MarketplacePlan,
+        MarketplaceSubscription,
+        MarketplaceSubscriptionProperties,
+        MarketplaceSubscriptionResourceArmPaginatedResult,
+        MaterializationComputeResource,
+        MaterializationSettings,
+        MedianStoppingPolicy,
+        MLAssistConfiguration,
+        MLAssistConfigurationDisabled,
+        MLAssistConfigurationEnabled,
+        MLFlowModelJobInput,
+        MLFlowModelJobOutput,
+        MLTableData,
+        MLTableJobInput,
+        MLTableJobOutput,
+        ModelConfiguration,
+        ModelContainer,
+        ModelContainerProperties,
+        ModelContainerResourceArmPaginatedResult,
+        ModelDeprecationInfo,
+        ModelPackageInput,
+        ModelPerformanceMetricThresholdBase,
+        ModelPerformanceSignal,
+        ModelSettings,
+        ModelSku,
+        ModelVersion,
+        ModelVersionProperties,
+        ModelVersionResourceArmPaginatedResult,
+        MonitorComputeConfigurationBase,
+        MonitorComputeIdentityBase,
+        MonitorDefinition,
+        MonitorEmailNotificationSettings,
+        MonitoringDataSegment,
+        MonitoringFeatureFilterBase,
+        MonitoringInputDataBase,
+        MonitoringSignalBase,
+        MonitoringTarget,
+        MonitoringThreshold,
+        MonitoringWorkspaceConnection,
+        MonitorNotificationSettings,
+        MonitorServerlessSparkCompute,
+        Mpi,
+        NCrossValidations,
+        NlpFixedParameters,
+        NlpParameterSubspace,
+        NlpSweepSettings,
+        NlpVertical,
+        NlpVerticalFeaturizationSettings,
+        NlpVerticalLimitSettings,
+        Nodes,
+        NodeStateCounts,
+        NoneAuthTypeWorkspaceConnectionProperties,
+        NoneDatastoreCredentials,
+        NotebookAccessTokenResult,
+        NotebookPreparationError,
+        NotebookResourceInfo,
+        NotificationSetting,
+        NumericalDataDriftMetricThreshold,
+        NumericalDataQualityMetricThreshold,
+        NumericalPredictionDriftMetricThreshold,
+        OAuth2AuthTypeWorkspaceConnectionProperties,
+        Objective,
+        OneLakeArtifact,
+        OneLakeDatastore,
+        OnlineDeployment,
+        OnlineDeploymentProperties,
+        OnlineDeploymentTrackedResourceArmPaginatedResult,
+        OnlineEndpoint,
+        OnlineEndpointProperties,
+        OnlineEndpointTrackedResourceArmPaginatedResult,
+        OnlineInferenceConfiguration,
+        OnlineRequestSettings,
+        OnlineScaleSettings,
+        OpenAIEndpointDeploymentResourceProperties,
+        OpenAIEndpointResourceProperties,
+        Operation,
+        OperationDisplay,
+        OperationListResult,
+        OsPatchingStatus,
+        OutboundRule,
+        OutboundRuleBasicResource,
+        OutboundRuleListResult,
+        OutputPathAssetReference,
+        PackageInputPathBase,
+        PackageInputPathId,
+        PackageInputPathUrl,
+        PackageInputPathVersion,
+        PackageRequest,
+        PackageResponse,
+        PaginatedComputeResourcesList,
+        PartialBatchDeployment,
+        PartialBatchDeploymentPartialMinimalTrackedResourceWithProperties,
+        PartialJobBase,
+        PartialJobBasePartialResource,
+        PartialManagedServiceIdentity,
+        PartialMinimalTrackedResource,
+        PartialMinimalTrackedResourceWithIdentity,
+        PartialMinimalTrackedResourceWithSku,
+        PartialMinimalTrackedResourceWithSkuAndIdentity,
+        PartialNotificationSetting,
+        PartialRegistryPartialTrackedResource,
+        PartialSku,
+        Password,
+        PATAuthTypeWorkspaceConnectionProperties,
+        PendingUploadCredentialDto,
+        PendingUploadRequestDto,
+        PendingUploadResponseDto,
+        PersonalComputeInstanceSettings,
+        PipelineJob,
+        PoolEnvironmentConfiguration,
+        PoolModelConfiguration,
+        PoolStatus,
+        PredictionDriftMetricThresholdBase,
+        PredictionDriftMonitoringSignal,
+        PrivateEndpoint,
+        PrivateEndpointConnection,
+        PrivateEndpointConnectionListResult,
+        PrivateEndpointDestination,
+        PrivateEndpointOutboundRule,
+        PrivateEndpointResource,
+        PrivateLinkResource,
+        PrivateLinkResourceListResult,
+        PrivateLinkServiceConnectionState,
+        ProbeSettings,
+        ProgressMetrics,
+        PropertiesBase,
+        ProxyResource,
+        PyTorch,
+        QueueSettings,
+        QuotaBaseProperties,
+        QuotaUpdateParameters,
+        RandomSamplingAlgorithm,
+        Ray,
+        Recurrence,
+        RecurrenceSchedule,
+        RecurrenceTrigger,
+        RegenerateEndpointKeysRequest,
+        RegenerateServiceAccountKeyContent,
+        Registry,
+        RegistryListCredentialsResult,
+        RegistryPartialManagedServiceIdentity,
+        RegistryPrivateEndpointConnection,
+        RegistryPrivateLinkServiceConnectionState,
+        RegistryRegionArmDetails,
+        RegistryTrackedResourceArmPaginatedResult,
+        Regression,
+        RegressionModelPerformanceMetricThreshold,
+        RegressionTrainingSettings,
+        RequestConfiguration,
+        RequestLogging,
+        RequestMatchPattern,
+        ResizeSchema,
+        Resource,
+        ResourceBase,
+        ResourceConfiguration,
+        ResourceId,
+        ResourceName,
+        ResourceQuota,
+        RollingInputData,
+        Route,
+        SamplingAlgorithm,
+        SASAuthTypeWorkspaceConnectionProperties,
+        SASCredential,
+        SASCredentialDto,
+        SasDatastoreCredentials,
+        SasDatastoreSecrets,
+        ScaleSettings,
+        ScaleSettingsInformation,
+        Schedule,
+        ScheduleActionBase,
+        ScheduleBase,
+        ScheduleProperties,
+        ScheduleResourceArmPaginatedResult,
+        ScriptReference,
+        ScriptsToExecute,
+        Seasonality,
+        SecretConfiguration,
+        ServerlessComputeSettings,
+        ServerlessEndpoint,
+        ServerlessEndpointCapacityReservation,
+        ServerlessEndpointProperties,
+        ServerlessEndpointStatus,
+        ServerlessEndpointTrackedResourceArmPaginatedResult,
+        ServerlessInferenceEndpoint,
+        ServerlessOffer,
+        ServiceManagedResourcesSettings,
+        ServicePrincipalAuthTypeWorkspaceConnectionProperties,
+        ServicePrincipalDatastoreCredentials,
+        ServicePrincipalDatastoreSecrets,
+        ServiceTagDestination,
+        ServiceTagOutboundRule,
+        SetupScripts,
+        SharedPrivateLinkResource,
+        Sku,
+        SkuCapacity,
+        SkuResource,
+        SkuResourceArmPaginatedResult,
+        SkuSetting,
+        SparkJob,
+        SparkJobEntry,
+        SparkJobPythonEntry,
+        SparkJobScalaEntry,
+        SparkResourceConfiguration,
+        SpeechEndpointDeploymentResourceProperties,
+        SpeechEndpointResourceProperties,
+        SslConfiguration,
+        StackEnsembleSettings,
+        StaticInputData,
+        StatusMessage,
+        StorageAccountDetails,
+        SweepJob,
+        SweepJobLimits,
+        SynapseSpark,
+        SynapseSparkProperties,
+        SystemCreatedAcrAccount,
+        SystemCreatedStorageAccount,
+        SystemData,
+        SystemService,
+        TableFixedParameters,
+        TableParameterSubspace,
+        TableSweepSettings,
+        TableVertical,
+        TableVerticalFeaturizationSettings,
+        TableVerticalLimitSettings,
+        TargetLags,
+        TargetRollingWindowSize,
+        TargetUtilizationScaleSettings,
+        TensorFlow,
+        TextClassification,
+        TextClassificationMultilabel,
+        TextNer,
+        ThrottlingRule,
+        TmpfsOptions,
+        TopNFeaturesByAttribution,
+        TrackedResource,
+        TrainingSettings,
+        TrialComponent,
+        TriggerBase,
+        TriggerOnceRequest,
+        TriggerRunSubmissionDto,
+        TritonInferencingServer,
+        TritonModelJobInput,
+        TritonModelJobOutput,
+        TruncationSelectionPolicy,
+        UpdateWorkspaceQuotas,
+        UpdateWorkspaceQuotasResult,
+        UriFileDataVersion,
+        UriFileJobInput,
+        UriFileJobOutput,
+        UriFolderDataVersion,
+        UriFolderJobInput,
+        UriFolderJobOutput,
+        Usage,
+        UsageName,
+        UserAccountCredentials,
+        UserAssignedIdentity,
+        UserCreatedAcrAccount,
+        UserCreatedStorageAccount,
+        UserIdentity,
+        UsernamePasswordAuthTypeWorkspaceConnectionProperties,
+        VirtualMachine,
+        VirtualMachineImage,
+        VirtualMachineSchema,
+        VirtualMachineSchemaProperties,
+        VirtualMachineSecrets,
+        VirtualMachineSecretsSchema,
+        VirtualMachineSize,
+        VirtualMachineSizeListResult,
+        VirtualMachineSshCredentials,
+        VolumeDefinition,
+        VolumeOptions,
+        Webhook,
+        Workspace,
+        WorkspaceConnectionAccessKey,
+        WorkspaceConnectionApiKey,
+        WorkspaceConnectionManagedIdentity,
+        WorkspaceConnectionOAuth2,
+        WorkspaceConnectionPersonalAccessToken,
+        WorkspaceConnectionPropertiesV2,
+        WorkspaceConnectionPropertiesV2BasicResource,
+        WorkspaceConnectionPropertiesV2BasicResourceArmPaginatedResult,
+        WorkspaceConnectionServicePrincipal,
+        WorkspaceConnectionSharedAccessSignature,
+        WorkspaceConnectionUpdateParameter,
+        WorkspaceConnectionUsernamePassword,
+        WorkspaceHubConfig,
+        WorkspaceListResult,
+        WorkspacePrivateEndpointResource,
+        WorkspaceUpdateParameters,
+    )
 except (SyntaxError, ImportError):
     from ._models import AADAuthTypeWorkspaceConnectionProperties  # type: ignore
     from ._models import AKS  # type: ignore
@@ -1368,8 +1370,8 @@
     DataCollectionMode,
     DataImportSourceType,
     DataReferenceCredentialType,
-    DataType,
     DatastoreType,
+    DataType,
     DefaultResourceProvisioningState,
     DeploymentModelVersionUpgradeOption,
     DeploymentProvisioningState,
@@ -1422,15 +1424,15 @@
     LogTrainingMetrics,
     LogValidationLoss,
     LogVerbosity,
-    MLAssistConfigurationType,
-    MLFlowAutologgerState,
     ManagedNetworkStatus,
     ManagedServiceIdentityType,
     MarketplaceSubscriptionProvisioningState,
     MarketplaceSubscriptionStatus,
     MaterializationStoreType,
     MediaType,
+    MLAssistConfigurationType,
     MlflowAutologger,
+    MLFlowAutologgerState,
     ModelLifecycleStatus,
     ModelProvider,
     ModelSize,
@@ -1528,10 +1530,10 @@
     UnitOfMeasure,
     UsageUnit,
     UseStl,
-    VMPriceOSType,
-    VMTier,
     ValidationMetricType,
+    VMPriceOSType,
     VmPriority,
+    VMTier,
     VolumeDefinitionType,
     WebhookType,
     WeekDay,
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_azure_machine_learning_workspaces_enums.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_azure_machine_learning_workspaces_enums.py
index 6ee730125ffe..fc1de0dae748 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_azure_machine_learning_workspaces_enums.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_azure_machine_learning_workspaces_enums.py
@@ -7,7 +7,9 @@
 # --------------------------------------------------------------------------
 
 from enum import Enum
+
 from six import with_metaclass
+
 from azure.core import CaseInsensitiveEnumMeta
 
 
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_models.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_models.py
index 1113e32d55cd..72261ed2910c 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_models.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_models.py
@@ -6,9 +6,10 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-from azure.core.exceptions import HttpResponseError
 import msrest.serialization
 
+from azure.core.exceptions import HttpResponseError
+
 
 class WorkspaceConnectionPropertiesV2(msrest.serialization.Model):
     """WorkspaceConnectionPropertiesV2.
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_models_py3.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_models_py3.py
index 967e3b42a8d4..d967e0c7e151 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_models_py3.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/models/_models_py3.py
@@ -9,9 +9,10 @@
 import datetime
 from typing import Any, Dict, List, Optional, Union
 
-from azure.core.exceptions import HttpResponseError
 import msrest.serialization
 
+from azure.core.exceptions import HttpResponseError
+
 from ._azure_machine_learning_workspaces_enums import *
 
 
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/__init__.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/__init__.py
index aefdb5620dd3..ed0728892746 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/__init__.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/__init__.py
@@ -6,61 +6,61 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-from ._usages_operations import UsagesOperations
-from ._virtual_machine_sizes_operations import VirtualMachineSizesOperations
-from ._quotas_operations import QuotasOperations
-from ._compute_operations import ComputeOperations
-from ._registries_operations import RegistriesOperations
-from ._workspace_features_operations import WorkspaceFeaturesOperations
-from ._capacity_reservation_groups_operations import CapacityReservationGroupsOperations
-from ._registry_code_containers_operations import RegistryCodeContainersOperations
-from ._registry_code_versions_operations import RegistryCodeVersionsOperations
-from ._registry_component_containers_operations import RegistryComponentContainersOperations
-from ._registry_component_versions_operations import RegistryComponentVersionsOperations
-from ._registry_data_containers_operations import RegistryDataContainersOperations
-from ._registry_data_versions_operations import RegistryDataVersionsOperations
-from ._registry_data_references_operations import RegistryDataReferencesOperations
-from ._registry_environment_containers_operations import RegistryEnvironmentContainersOperations
-from ._registry_environment_versions_operations import RegistryEnvironmentVersionsOperations
-from ._marketplace_subscriptions_operations import MarketplaceSubscriptionsOperations
-from ._registry_model_containers_operations import RegistryModelContainersOperations
-from ._registry_model_versions_operations import RegistryModelVersionsOperations
-from ._batch_endpoints_operations import BatchEndpointsOperations
 from ._batch_deployments_operations import BatchDeploymentsOperations
+from ._batch_endpoints_operations import BatchEndpointsOperations
+from ._capacity_reservation_groups_operations import CapacityReservationGroupsOperations
 from ._code_containers_operations import CodeContainersOperations
 from ._code_versions_operations import CodeVersionsOperations
 from ._component_containers_operations import ComponentContainersOperations
 from ._component_versions_operations import ComponentVersionsOperations
+from ._compute_operations import ComputeOperations
 from ._data_containers_operations import DataContainersOperations
 from ._data_versions_operations import DataVersionsOperations
 from ._datastores_operations import DatastoresOperations
+from ._endpoint_deployment_operations import EndpointDeploymentOperations
+from ._endpoint_operations import EndpointOperations
 from ._environment_containers_operations import EnvironmentContainersOperations
 from ._environment_versions_operations import EnvironmentVersionsOperations
-from ._featureset_containers_operations import FeaturesetContainersOperations
 from ._features_operations import FeaturesOperations
+from ._featureset_containers_operations import FeaturesetContainersOperations
 from ._featureset_versions_operations import FeaturesetVersionsOperations
 from ._featurestore_entity_containers_operations import FeaturestoreEntityContainersOperations
 from ._featurestore_entity_versions_operations import FeaturestoreEntityVersionsOperations
-from ._inference_pools_operations import InferencePoolsOperations
 from ._inference_endpoints_operations import InferenceEndpointsOperations
 from ._inference_groups_operations import InferenceGroupsOperations
+from ._inference_pools_operations import InferencePoolsOperations
 from ._jobs_operations import JobsOperations
 from ._labeling_jobs_operations import LabelingJobsOperations
+from ._managed_network_provisions_operations import ManagedNetworkProvisionsOperations
+from ._managed_network_settings_rule_operations import ManagedNetworkSettingsRuleOperations
+from ._marketplace_subscriptions_operations import MarketplaceSubscriptionsOperations
 from ._model_containers_operations import ModelContainersOperations
 from ._model_versions_operations import ModelVersionsOperations
-from ._online_endpoints_operations import OnlineEndpointsOperations
 from ._online_deployments_operations import OnlineDeploymentsOperations
-from ._schedules_operations import SchedulesOperations
-from ._serverless_endpoints_operations import ServerlessEndpointsOperations
+from ._online_endpoints_operations import OnlineEndpointsOperations
 from ._operations import Operations
-from ._workspaces_operations import WorkspacesOperations
-from ._workspace_connections_operations import WorkspaceConnectionsOperations
-from ._endpoint_deployment_operations import EndpointDeploymentOperations
-from ._endpoint_operations import EndpointOperations
-from ._managed_network_settings_rule_operations import ManagedNetworkSettingsRuleOperations
 from ._private_endpoint_connections_operations import PrivateEndpointConnectionsOperations
 from ._private_link_resources_operations import PrivateLinkResourcesOperations
-from ._managed_network_provisions_operations import ManagedNetworkProvisionsOperations
+from ._quotas_operations import QuotasOperations
+from ._registries_operations import RegistriesOperations
+from ._registry_code_containers_operations import RegistryCodeContainersOperations
+from ._registry_code_versions_operations import RegistryCodeVersionsOperations
+from ._registry_component_containers_operations import RegistryComponentContainersOperations
+from ._registry_component_versions_operations import RegistryComponentVersionsOperations
+from ._registry_data_containers_operations import RegistryDataContainersOperations
+from ._registry_data_references_operations import RegistryDataReferencesOperations
+from ._registry_data_versions_operations import RegistryDataVersionsOperations
+from ._registry_environment_containers_operations import RegistryEnvironmentContainersOperations
+from ._registry_environment_versions_operations import RegistryEnvironmentVersionsOperations
+from ._registry_model_containers_operations import RegistryModelContainersOperations
+from ._registry_model_versions_operations import RegistryModelVersionsOperations
+from ._schedules_operations import SchedulesOperations
+from ._serverless_endpoints_operations import ServerlessEndpointsOperations
+from ._usages_operations import UsagesOperations
+from ._virtual_machine_sizes_operations import VirtualMachineSizesOperations
+from ._workspace_connections_operations import WorkspaceConnectionsOperations
+from ._workspace_features_operations import WorkspaceFeaturesOperations
+from ._workspaces_operations import WorkspacesOperations
 
 __all__ = [
     'UsagesOperations',
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_code_versions_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_code_versions_operations.py
index 2987c7e0f3e8..e6ae340cc4ef 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_code_versions_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_code_versions_operations.py
@@ -10,7 +10,13 @@
 
 from msrest import Serializer
 
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.paging import ItemPaged
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import HttpResponse
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_component_versions_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_component_versions_operations.py
index e905e88dd99e..dbac04004b2a 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_component_versions_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_component_versions_operations.py
@@ -10,7 +10,13 @@
 
 from msrest import Serializer
 
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.paging import ItemPaged
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import HttpResponse
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_data_versions_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_data_versions_operations.py
index e9d49420dbc3..f6ddca244ba7 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_data_versions_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_data_versions_operations.py
@@ -10,7 +10,13 @@
 
 from msrest import Serializer
 
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.paging import ItemPaged
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import HttpResponse
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_environment_versions_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_environment_versions_operations.py
index ae386e0b583e..5b700ab6a1a4 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_environment_versions_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_environment_versions_operations.py
@@ -10,7 +10,13 @@
 
 from msrest import Serializer
 
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.paging import ItemPaged
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import HttpResponse
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_jobs_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_jobs_operations.py
index 6a369791d7b5..8f178960a4fa 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_jobs_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_jobs_operations.py
@@ -10,7 +10,13 @@
 
 from msrest import Serializer
 
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.paging import ItemPaged
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import HttpResponse
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_model_versions_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_model_versions_operations.py
index fa9c290bfd88..09db84bda340 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_model_versions_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_model_versions_operations.py
@@ -10,7 +10,13 @@
 
 from msrest import Serializer
 
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.paging import ItemPaged
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import HttpResponse
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_registry_data_references_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_registry_data_references_operations.py
index e2f8b80b3fe0..fc7a41ea4354 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_registry_data_references_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_registry_data_references_operations.py
@@ -10,7 +10,13 @@
 
 from msrest import Serializer
 
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import HttpResponse
 from azure.core.rest import HttpRequest
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_schedules_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_schedules_operations.py
index 1f04079eaa40..f996d8189119 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_schedules_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_schedules_operations.py
@@ -10,7 +10,13 @@
 
 from msrest import Serializer
 
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.paging import ItemPaged
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import HttpResponse
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_serverless_endpoints_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_serverless_endpoints_operations.py
index b653c926b827..0f309e3e5a1e 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_serverless_endpoints_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/operations/_serverless_endpoints_operations.py
@@ -10,7 +10,13 @@
 
 from msrest import Serializer
 
-from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceExistsError,
+    ResourceNotFoundError,
+    map_error,
+)
 from azure.core.paging import ItemPaged
 from azure.core.pipeline import PipelineResponse
 from azure.core.pipeline.transport import HttpResponse
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/__init__.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/__init__.py
index b0bcfc0881ac..efc3858e166d 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/__init__.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/__init__.py
@@ -165,11 +165,11 @@
     FeatureAttributionDriftSignal,
     GenerationSafetyQualitySignal,
     LlmData,
+    ModelPerformanceSignal,
     MonitorFeatureFilter,
     PredictionDriftSignal,
     ProductionData,
     ReferenceData,
-    ModelPerformanceSignal,
 )
 from ._monitoring.target import MonitoringTarget
 from ._monitoring.thresholds import (
@@ -181,11 +181,11 @@
     DataQualityMetricThreshold,
     FeatureAttributionDriftMetricThreshold,
     GenerationSafetyQualityMonitoringMetricThreshold,
-    NumericalDriftMetrics,
-    PredictionDriftMetricThreshold,
-    ModelPerformanceMetricThreshold,
     ModelPerformanceClassificationThresholds,
+    ModelPerformanceMetricThreshold,
     ModelPerformanceRegressionThresholds,
+    NumericalDriftMetrics,
+    PredictionDriftMetricThreshold,
 )
 from ._notification.notification import Notification
 from ._registry.registry import Registry
@@ -195,7 +195,7 @@
     SystemCreatedStorageAccount,
 )
 from ._resource import Resource
-from ._schedule.schedule import JobSchedule, Schedule
+from ._schedule.schedule import JobSchedule, Schedule, ScheduleTriggerResult
 from ._schedule.trigger import CronTrigger, RecurrencePattern, RecurrenceTrigger
 from ._system_data import SystemData
 from ._validation import ValidationResult
@@ -204,8 +204,8 @@
 from ._workspace.connections.workspace_connection_subtypes import (
     AzureAISearchWorkspaceConnection,
     AzureAIServiceWorkspaceConnection,
-    AzureOpenAIWorkspaceConnection,
     AzureBlobStoreWorkspaceConnection,
+    AzureOpenAIWorkspaceConnection,
 )
 from ._workspace.customer_managed_key import CustomerManagedKey
 from ._workspace.diagnose import (
@@ -349,6 +349,7 @@
     "JobSchedule",
     "ImportDataSchedule",
     "Schedule",
+    "ScheduleTriggerResult",
     "ComputePowerAction",
     "ComputeSchedules",
     "ComputeStartStopSchedule",
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_component/component.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_component/component.py
index 525ff300b1cd..8654103af6d2 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_component/component.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_component/component.py
@@ -9,7 +9,7 @@
 
 from marshmallow import INCLUDE
 
-from ..._restclient.v2023_08_01_preview.models import (
+from ..._restclient.v2024_01_01_preview.models import (
     ComponentContainer,
     ComponentContainerProperties,
     ComponentVersion,
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/job.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/job.py
index e169dce57435..aca3bb70515a 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/job.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/job.py
@@ -16,7 +16,7 @@
 from azure.ai.ml._restclient.runhistory.models import Run
 from azure.ai.ml._restclient.v2023_04_01_preview.models import JobBase, JobService
 from azure.ai.ml._restclient.v2023_04_01_preview.models import JobType as RestJobType
-from azure.ai.ml._restclient.v2023_08_01_preview.models import JobBase as JobBase_2308
+from azure.ai.ml._restclient.v2024_01_01_preview.models import JobBase as JobBase_2401
 from azure.ai.ml._utils._html_utils import make_link, to_html
 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, CommonYamlFields
@@ -282,7 +282,7 @@ def _load(
 
     @classmethod
     def _from_rest_object(  # pylint: disable=too-many-return-statements
-        cls, obj: Union[JobBase, JobBase_2308, Run]
+        cls, obj: Union[JobBase, JobBase_2401, Run]
     ) -> "Job":  # pylint: disable=too-many-return-statements
         from azure.ai.ml.entities import PipelineJob
         from azure.ai.ml.entities._builders.command import Command
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/pipeline/pipeline_job.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/pipeline/pipeline_job.py
index 44a7d490753d..49e3bd3f0921 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/pipeline/pipeline_job.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/pipeline/pipeline_job.py
@@ -12,8 +12,8 @@
 
 from typing_extensions import Literal
 
-from azure.ai.ml._restclient.v2023_08_01_preview.models import JobBase
-from azure.ai.ml._restclient.v2023_08_01_preview.models import PipelineJob as RestPipelineJob
+from azure.ai.ml._restclient.v2024_01_01_preview.models import JobBase
+from azure.ai.ml._restclient.v2024_01_01_preview.models import PipelineJob as RestPipelineJob
 from azure.ai.ml._schema import PathAwareSchema
 from azure.ai.ml._schema.pipeline.pipeline_job import PipelineJobSchema
 from azure.ai.ml._utils._arm_id_utils import get_resource_name_from_arm_id_safe
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_schedule/schedule.py b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_schedule/schedule.py
index 08556c8fdb61..5632ad86a385 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_schedule/schedule.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/entities/_schedule/schedule.py
@@ -16,6 +16,7 @@
 from azure.ai.ml._restclient.v2023_06_01_preview.models import Schedule as RestSchedule
 from azure.ai.ml._restclient.v2023_06_01_preview.models import ScheduleActionType as RestScheduleActionType
 from azure.ai.ml._restclient.v2023_06_01_preview.models import ScheduleProperties
+from azure.ai.ml._restclient.v2024_01_01_preview.models import TriggerRunSubmissionDto as RestTriggerRunSubmissionDto
 from azure.ai.ml._schema.schedule.schedule import JobScheduleSchema
 from azure.ai.ml._utils.utils import camel_to_snake, dump_yaml_to_file, is_private_preview_enabled
 from azure.ai.ml.constants import JobType
@@ -117,12 +118,18 @@ def _resolve_cls_and_type(
 
     @property
     def create_job(self) -> Any:  # pylint: disable=useless-return
+        """The create_job entity associated with the schedule if exists."""
         module_logger.warning("create_job is not a valid property of %s", str(type(self)))
         # return None here just to be explicit
         return None
 
     @create_job.setter
     def create_job(self, value: Any) -> None:  # pylint: disable=unused-argument
+        """Set the create_job entity associated with the schedule if exists.
+
+        :param value: The create_job entity associated with the schedule if exists.
+        :type value: Any
+        """
         module_logger.warning("create_job is not a valid property of %s", str(type(self)))
 
     @property
@@ -470,3 +477,28 @@ def _get_telemetry_values(self, *args: Any, **kwargs: Any) -> Dict[Literal["trig
         :rtype: Dict[Literal["trigger_type"], str]
         """
         return {"trigger_type": type(self.trigger).__name__}
+
+
+class ScheduleTriggerResult:
+    """Schedule trigger result returned by trigger an enabled schedule once.
+
+    This class shouldn't be instantiated directly. Instead, it is used as the return type of schedule trigger.
+    """
+
+    def __init__(self, **kwargs):  # pylint: disable=unused-argument
+        self.job_name = kwargs.get("job_name", None)
+        self.schedule_action_type = kwargs.get("schedule_action_type", None)
+
+    @classmethod
+    def _from_rest_object(cls, obj: RestTriggerRunSubmissionDto) -> "ScheduleTriggerResult":
+        """Construct a ScheduleJob from a rest object.
+
+        :param obj: The rest object to construct from.
+        :type obj: ~azure.ai.ml._restclient.v2024_01_01_preview.models.TriggerRunSubmissionDto
+        :return: The constructed ScheduleJob.
+        :rtype: ScheduleTriggerResult
+        """
+        return cls(
+            schedule_action_type=obj.schedule_action_type,
+            job_name=obj.submission_id,
+        )
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_component_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_component_operations.py
index 7916c2feac9f..9f1fe24b505b 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_component_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_component_operations.py
@@ -14,8 +14,8 @@
 from azure.ai.ml._restclient.v2021_10_01_dataplanepreview import (
     AzureMachineLearningWorkspaces as ServiceClient102021Dataplane,
 )
-from azure.ai.ml._restclient.v2022_10_01 import AzureMachineLearningWorkspaces as ServiceClient102022
-from azure.ai.ml._restclient.v2022_10_01.models import ComponentVersion, ListViewType
+from azure.ai.ml._restclient.v2024_01_01_preview import AzureMachineLearningWorkspaces as ServiceClient012024
+from azure.ai.ml._restclient.v2024_01_01_preview.models import ComponentVersion, ListViewType
 from azure.ai.ml._scope_dependent_operations import (
     OperationConfig,
     OperationsContainer,
@@ -91,7 +91,7 @@ def __init__(
         self,
         operation_scope: OperationScope,
         operation_config: OperationConfig,
-        service_client: Union[ServiceClient102022, ServiceClient102021Dataplane],
+        service_client: Union[ServiceClient012024, ServiceClient102021Dataplane],
         all_operations: OperationsContainer,
         preflight_operation: Optional[DeploymentsOperations] = None,
         **kwargs: Dict,
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_job_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_job_operations.py
index 188cad09515a..10ad24014a0f 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_job_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_job_operations.py
@@ -29,8 +29,8 @@
 from azure.ai.ml._restclient.runhistory.models import Run
 from azure.ai.ml._restclient.v2023_04_01_preview import AzureMachineLearningWorkspaces as ServiceClient022023Preview
 from azure.ai.ml._restclient.v2023_04_01_preview.models import JobBase, ListViewType, UserIdentity
-from azure.ai.ml._restclient.v2023_08_01_preview.models import JobBase as JobBase_2308
 from azure.ai.ml._restclient.v2023_08_01_preview.models import JobType as RestJobType
+from azure.ai.ml._restclient.v2024_01_01_preview.models import JobBase as JobBase_2401
 from azure.ai.ml._scope_dependent_operations import (
     OperationConfig,
     OperationsContainer,
@@ -175,7 +175,7 @@ def __init__(
             self._all_operations, self._operation_scope, self._operation_config
         )  # pylint: disable=line-too-long
 
-        self.service_client_08_2023_preview = kwargs.pop("service_client_08_2023_preview", None)
+        self.service_client_01_2024_preview = kwargs.pop("service_client_01_2024_preview", None)
         self._kwargs = kwargs
 
         self._requests_pipeline: HttpPipeline = kwargs.pop("requests_pipeline")
@@ -695,7 +695,7 @@ def create_or_update(
             # request for submitting to ES. Once we request to ES and start the run, we
             # need to put the same body to MFE to append user tags etc.
             if rest_job_resource.properties.job_type == RestJobType.PIPELINE:
-                job_object = self._get_job_2308(rest_job_resource.name)
+                job_object = self._get_job_2401(rest_job_resource.name)
             else:
                 job_object = self._get_job(rest_job_resource.name)
             if result.properties.tags is not None:
@@ -720,10 +720,10 @@ def _create_or_update_with_different_version_api(  # pylint: disable=name-too-lo
         service_client_operation = self._operation_2023_02_preview
         # Upgrade api from 2023-04-01-preview to 2023-08-01 for pipeline job
         if rest_job_resource.properties.job_type == RestJobType.PIPELINE:
-            service_client_operation = self.service_client_08_2023_preview.jobs
+            service_client_operation = self.service_client_01_2024_preview.jobs
 
         if rest_job_resource.properties.job_type == RestJobType.SWEEP:
-            service_client_operation = self.service_client_08_2023_preview.jobs
+            service_client_operation = self.service_client_01_2024_preview.jobs
 
         result = service_client_operation.create_or_update(
             id=rest_job_resource.name,
@@ -738,7 +738,7 @@ def _create_or_update_with_different_version_api(  # pylint: disable=name-too-lo
     def _archive_or_restore(self, name: str, is_archived: bool) -> None:
         job_object = self._get_job(name)
         if job_object.properties.job_type == RestJobType.PIPELINE:
-            job_object = self._get_job_2308(name)
+            job_object = self._get_job_2401(name)
         if _is_pipeline_child_job(job_object):
             raise PipelineChildJobError(job_id=job_object.id)
         job_object.properties.is_archived = is_archived
@@ -1029,10 +1029,10 @@ def _get_job(self, name: str) -> JobBase:
             **self._kwargs,
         )
 
-    # Upgrade api from 2023-04-01-preview to 2023-08-01 for pipeline job
-    # We can remove this function once `_get_job` function has also been upgraded to 2023-08-01 api
-    def _get_job_2308(self, name: str) -> JobBase_2308:
-        service_client_operation = self.service_client_08_2023_preview.jobs
+    # Upgrade api from 2023-04-01-preview to 2024-01-01-preview for pipeline job
+    # We can remove this function once `_get_job` function has also been upgraded to the same version with pipeline
+    def _get_job_2401(self, name: str) -> JobBase_2401:
+        service_client_operation = self.service_client_01_2024_preview.jobs
         return service_client_operation.get(
             id=name,
             resource_group_name=self._operation_scope.resource_group_name,
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_schedule_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_schedule_operations.py
index 34d304876711..4afc4b383fef 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_schedule_operations.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_schedule_operations.py
@@ -1,10 +1,12 @@
 # ---------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
-# pylint: disable=protected-access, too-many-nested-blocks
+# pylint: disable=protected-access
+from datetime import datetime, timezone
 from typing import Any, Iterable, List, Optional, Tuple, cast
 
 from azure.ai.ml._restclient.v2023_06_01_preview import AzureMachineLearningWorkspaces as ServiceClient062023Preview
+from azure.ai.ml._restclient.v2024_01_01_preview import AzureMachineLearningWorkspaces as ServiceClient012024Preview
 from azure.ai.ml._scope_dependent_operations import (
     OperationConfig,
     OperationsContainer,
@@ -24,6 +26,7 @@
 from azure.core.tracing.decorator import distributed_trace
 
 from .._restclient.v2022_10_01.models import ScheduleListViewType
+from .._restclient.v2024_01_01_preview.models import TriggerOnceRequest
 from .._utils._arm_id_utils import AMLNamedArmId, AMLVersionedArmId, is_ARM_id_for_parented_resource
 from .._utils._azureml_polling import AzureMLPolling
 from .._utils.utils import snake_to_camel
@@ -44,6 +47,7 @@
     MonitorDatasetContext,
     MonitorSignalType,
 )
+from ..entities._schedule.schedule import ScheduleTriggerResult
 from . import DataOperations, JobOperations, OnlineDeploymentOperations
 from ._job_ops_helper import stream_logs_until_completion
 from ._operation_orchestrator import OperationOrchestrator
@@ -66,6 +70,7 @@ def __init__(
         operation_scope: OperationScope,
         operation_config: OperationConfig,
         service_client_06_2023_preview: ServiceClient062023Preview,
+        service_client_01_2024_preview: ServiceClient012024Preview,
         all_operations: OperationsContainer,
         credential: TokenCredential,
         **kwargs: Any,
@@ -73,6 +78,9 @@ def __init__(
         super(ScheduleOperations, self).__init__(operation_scope, operation_config)
         ops_logger.update_info(kwargs)
         self.service_client = service_client_06_2023_preview.schedules
+        # Note: Trigger once is supported since 24_01, we don't upgrade other operations' client because there are
+        # some breaking changes, for example: AzMonMonitoringAlertNotificationSettings is removed.
+        self.schedule_trigger_service_client = service_client_01_2024_preview.schedules
         self._all_operations = all_operations
         self._stream_logs_until_completion = stream_logs_until_completion
         # Dataplane service clients are lazily created as they are needed
@@ -244,7 +252,6 @@ def begin_create_or_update(
             self._resolve_monitor_schedule_arm_id(schedule)
         # Create schedule
         schedule_data = schedule._to_rest_object()  # type: ignore
-        print(schedule_data.properties.tags)
         poller = self.service_client.begin_create_or_update(
             resource_group_name=self._operation_scope.resource_group_name,
             workspace_name=self._workspace_name,
@@ -293,6 +300,30 @@ def begin_disable(
         schedule._is_enabled = False
         return self.begin_create_or_update(schedule, **kwargs)
 
+    @distributed_trace
+    @monitor_with_activity(logger, "Schedule.Trigger", ActivityType.PUBLICAPI)
+    def trigger(
+        self,
+        name: str,
+        **kwargs: Any,
+    ) -> ScheduleTriggerResult:
+        """Trigger a schedule once.
+
+        :param name: Schedule name.
+        :type name: str
+        :return: TriggerRunSubmissionDto, or the result of cls(response)
+        :rtype: ~azure.ai.ml.entities.ScheduleTriggerResult
+        """
+        schedule_time = kwargs.pop("schedule_time", datetime.now(timezone.utc).isoformat())
+        return self.schedule_trigger_service_client.trigger(
+            name=name,
+            resource_group_name=self._operation_scope.resource_group_name,
+            workspace_name=self._workspace_name,
+            body=TriggerOnceRequest(schedule_time=schedule_time),
+            cls=lambda _, obj, __: ScheduleTriggerResult._from_rest_object(obj),
+            **kwargs,
+        )
+
     def _resolve_monitor_schedule_arm_id(  # pylint:disable=too-many-branches,too-many-statements
         self, schedule: MonitorSchedule
     ) -> None:
diff --git a/sdk/ml/azure-ai-ml/tests/dsl/unittests/test_dsl_pipeline.py b/sdk/ml/azure-ai-ml/tests/dsl/unittests/test_dsl_pipeline.py
index 9d96b2eb7750..f7f11ba709ae 100644
--- a/sdk/ml/azure-ai-ml/tests/dsl/unittests/test_dsl_pipeline.py
+++ b/sdk/ml/azure-ai-ml/tests/dsl/unittests/test_dsl_pipeline.py
@@ -8,6 +8,9 @@
 
 import pydash
 import pytest
+from test_configs.dsl_pipeline import data_binding_expression
+from test_utilities.utils import assert_job_cancel, omit_with_wildcard, prepare_dsl_curated
+
 from azure.ai.ml import (
     AmlTokenConfiguration,
     Input,
@@ -50,8 +53,6 @@
     UserErrorException,
     ValidationException,
 )
-from test_configs.dsl_pipeline import data_binding_expression
-from test_utilities.utils import assert_job_cancel, omit_with_wildcard, prepare_dsl_curated
 
 from .._util import _DSL_TIMEOUT_SECOND, get_predecessors
 
@@ -959,7 +960,7 @@ def mock_from_rest(*args, **kwargs):
             side_effect=mock_arm_id,
         ):
             with mock.patch(
-                "azure.ai.ml._restclient.v2022_10_01.operations.ComponentVersionsOperations.create_or_update",
+                "azure.ai.ml._restclient.v2024_01_01_preview.operations.ComponentVersionsOperations.create_or_update",
                 side_effect=mock_create,
             ):
                 with mock.patch.object(Component, "_from_rest_object", side_effect=mock_from_rest):
diff --git a/sdk/ml/azure-ai-ml/tests/schedule/e2etests/test_schedule.py b/sdk/ml/azure-ai-ml/tests/schedule/e2etests/test_schedule.py
index 628a44f8ce32..bd617a09bf80 100644
--- a/sdk/ml/azure-ai-ml/tests/schedule/e2etests/test_schedule.py
+++ b/sdk/ml/azure-ai-ml/tests/schedule/e2etests/test_schedule.py
@@ -6,7 +6,7 @@
 
 from azure.ai.ml import MLClient
 from azure.ai.ml.constants._common import LROConfigurations
-from azure.ai.ml.entities import AmlTokenConfiguration, CronTrigger, PipelineJob
+from azure.ai.ml.entities import AmlTokenConfiguration, CronTrigger, PipelineJob, ScheduleTriggerResult
 from azure.ai.ml.entities._load_functions import load_job, load_schedule
 
 from .._util import _SCHEDULE_TIMEOUT_SECOND, TRIGGER_ENDTIME, TRIGGER_ENDTIME_DICT
@@ -41,6 +41,10 @@ def test_schedule_lifetime(self, client: MLClient, randstr: Callable[[], str]):
         assert rest_schedule._is_enabled is True
         job: PipelineJob = rest_schedule.create_job
         assert isinstance(job.identity, AmlTokenConfiguration)
+        # trigger once
+        result: ScheduleTriggerResult = client.schedules.trigger(schedule.name, schedule_time="2024-02-19T00:00:00")
+        assert result.job_name is not None
+        assert client.jobs.get(name=result.job_name) is not None
         # disable
         rest_schedule = client.schedules.begin_disable(schedule.name).result(timeout=LROConfigurations.POLLING_TIMEOUT)
         assert rest_schedule._is_enabled is False