Support schedule trigger action

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.

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"
 }

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,14 +587,15 @@ 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)
         self._schedules = ScheduleOperations(
             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,

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

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

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',

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]]
 

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]]