Skip to content

Commit 7c98196

Browse files
brynn-codebrynn-code
authored
Support schedule trigger action (#33918)
* Update 2024-01-01-preview restclient Signed-off-by: Brynn Yin <[email protected]> * Upgrade pipeline job and component to 24-01 api Signed-off-by: Brynn Yin <[email protected]> * Add trigger api to operation Signed-off-by: Brynn Yin <[email protected]> * Revert mfe.json change Signed-off-by: Brynn Yin <[email protected]> * Update class name of trigger result Signed-off-by: Brynn Yin <[email protected]> * Use 24client only for schedule trigger Signed-off-by: Brynn Yin <[email protected]> * Fix test mock Signed-off-by: Brynn Yin <[email protected]> * Fix analyze Signed-off-by: Brynn Yin <[email protected]> --------- Signed-off-by: Brynn Yin <[email protected]>
1 parent 46ab212 commit 7c98196

39 files changed

+1157
-790
lines changed

‎sdk/ml/azure-ai-ml/CHANGELOG.md‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
- Workspace Connection has new read-only subclass: AzureBlobStoreWorkspaceConnectionSchema.
88
- Workspace Connection supports 2 new types under main class: gen 2 and azure_one_lake.
99
- Workspace Connection LIST operation can return data connections via new optional flag: include_data_connections.
10+
- Support `ml_client.schedules.trigger(name='my_schedule')` function to trigger a schedule once.
1011

1112
### Bugs Fixed
1213
- Fix pipeline job `outputs` not load correctly when `component: <local-file>` exists in pipeline job yaml.

‎sdk/ml/azure-ai-ml/assets.json‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,5 @@
22
"AssetsRepo": "Azure/azure-sdk-assets",
33
"AssetsRepoPrefixPath": "python",
44
"TagPrefix": "python/ml/azure-ai-ml",
5-
"Tag": "python/ml/azure-ai-ml_67cc2fc9de"
5+
"Tag": "python/ml/azure-ai-ml_1af7fe2f76"
66
}

‎sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py‎

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -345,6 +345,13 @@ def __init__(
345345
**kwargs,
346346
)
347347

348+
self._service_client_01_2024_preview = ServiceClient012024Preview(
349+
credential=self._credential,
350+
subscription_id=self._operation_scope._subscription_id,
351+
base_url=base_url,
352+
**kwargs,
353+
)
354+
348355
# A general purpose, user-configurable pipeline for making
349356
# http requests
350357
self._requests_pipeline = HttpPipeline(**kwargs)
@@ -566,7 +573,7 @@ def __init__(
566573
self._components = ComponentOperations(
567574
self._operation_scope,
568575
self._operation_config,
569-
self._service_client_10_2021_dataplanepreview if registry_name else self._service_client_10_2022,
576+
self._service_client_10_2021_dataplanepreview if registry_name else self._service_client_01_2024_preview,
570577
self._operation_container,
571578
self._preflight,
572579
**ops_kwargs,
@@ -580,14 +587,15 @@ def __init__(
580587
self._credential,
581588
_service_client_kwargs=kwargs,
582589
requests_pipeline=self._requests_pipeline,
583-
service_client_08_2023_preview=self._service_client_08_2023_preview,
590+
service_client_01_2024_preview=self._service_client_01_2024_preview,
584591
**ops_kwargs,
585592
)
586593
self._operation_container.add(AzureMLResourceType.JOB, self._jobs)
587594
self._schedules = ScheduleOperations(
588595
self._operation_scope,
589596
self._operation_config,
590597
self._service_client_06_2023_preview,
598+
self._service_client_01_2024_preview,
591599
self._operation_container,
592600
self._credential,
593601
_service_client_kwargs=kwargs,

‎sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/_azure_machine_learning_workspaces.py‎

Lines changed: 57 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,63 @@
1515

1616
from . import models
1717
from ._configuration import AzureMachineLearningWorkspacesConfiguration
18-
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
18+
from .operations import (
19+
BatchDeploymentsOperations,
20+
BatchEndpointsOperations,
21+
CapacityReservationGroupsOperations,
22+
CodeContainersOperations,
23+
CodeVersionsOperations,
24+
ComponentContainersOperations,
25+
ComponentVersionsOperations,
26+
ComputeOperations,
27+
DataContainersOperations,
28+
DatastoresOperations,
29+
DataVersionsOperations,
30+
EndpointDeploymentOperations,
31+
EndpointOperations,
32+
EnvironmentContainersOperations,
33+
EnvironmentVersionsOperations,
34+
FeaturesetContainersOperations,
35+
FeaturesetVersionsOperations,
36+
FeaturesOperations,
37+
FeaturestoreEntityContainersOperations,
38+
FeaturestoreEntityVersionsOperations,
39+
InferenceEndpointsOperations,
40+
InferenceGroupsOperations,
41+
InferencePoolsOperations,
42+
JobsOperations,
43+
LabelingJobsOperations,
44+
ManagedNetworkProvisionsOperations,
45+
ManagedNetworkSettingsRuleOperations,
46+
MarketplaceSubscriptionsOperations,
47+
ModelContainersOperations,
48+
ModelVersionsOperations,
49+
OnlineDeploymentsOperations,
50+
OnlineEndpointsOperations,
51+
Operations,
52+
PrivateEndpointConnectionsOperations,
53+
PrivateLinkResourcesOperations,
54+
QuotasOperations,
55+
RegistriesOperations,
56+
RegistryCodeContainersOperations,
57+
RegistryCodeVersionsOperations,
58+
RegistryComponentContainersOperations,
59+
RegistryComponentVersionsOperations,
60+
RegistryDataContainersOperations,
61+
RegistryDataReferencesOperations,
62+
RegistryDataVersionsOperations,
63+
RegistryEnvironmentContainersOperations,
64+
RegistryEnvironmentVersionsOperations,
65+
RegistryModelContainersOperations,
66+
RegistryModelVersionsOperations,
67+
SchedulesOperations,
68+
ServerlessEndpointsOperations,
69+
UsagesOperations,
70+
VirtualMachineSizesOperations,
71+
WorkspaceConnectionsOperations,
72+
WorkspaceFeaturesOperations,
73+
WorkspacesOperations,
74+
)
1975

2076
if TYPE_CHECKING:
2177
# 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‎

Lines changed: 58 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
# --------------------------------------------------------------------------
88

99
from copy import deepcopy
10-
from typing import Any, Awaitable, TYPE_CHECKING
10+
from typing import TYPE_CHECKING, Any, Awaitable
1111

1212
from msrest import Deserializer, Serializer
1313

@@ -16,7 +16,63 @@
1616

1717
from .. import models
1818
from ._configuration import AzureMachineLearningWorkspacesConfiguration
19-
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
19+
from .operations import (
20+
BatchDeploymentsOperations,
21+
BatchEndpointsOperations,
22+
CapacityReservationGroupsOperations,
23+
CodeContainersOperations,
24+
CodeVersionsOperations,
25+
ComponentContainersOperations,
26+
ComponentVersionsOperations,
27+
ComputeOperations,
28+
DataContainersOperations,
29+
DatastoresOperations,
30+
DataVersionsOperations,
31+
EndpointDeploymentOperations,
32+
EndpointOperations,
33+
EnvironmentContainersOperations,
34+
EnvironmentVersionsOperations,
35+
FeaturesetContainersOperations,
36+
FeaturesetVersionsOperations,
37+
FeaturesOperations,
38+
FeaturestoreEntityContainersOperations,
39+
FeaturestoreEntityVersionsOperations,
40+
InferenceEndpointsOperations,
41+
InferenceGroupsOperations,
42+
InferencePoolsOperations,
43+
JobsOperations,
44+
LabelingJobsOperations,
45+
ManagedNetworkProvisionsOperations,
46+
ManagedNetworkSettingsRuleOperations,
47+
MarketplaceSubscriptionsOperations,
48+
ModelContainersOperations,
49+
ModelVersionsOperations,
50+
OnlineDeploymentsOperations,
51+
OnlineEndpointsOperations,
52+
Operations,
53+
PrivateEndpointConnectionsOperations,
54+
PrivateLinkResourcesOperations,
55+
QuotasOperations,
56+
RegistriesOperations,
57+
RegistryCodeContainersOperations,
58+
RegistryCodeVersionsOperations,
59+
RegistryComponentContainersOperations,
60+
RegistryComponentVersionsOperations,
61+
RegistryDataContainersOperations,
62+
RegistryDataReferencesOperations,
63+
RegistryDataVersionsOperations,
64+
RegistryEnvironmentContainersOperations,
65+
RegistryEnvironmentVersionsOperations,
66+
RegistryModelContainersOperations,
67+
RegistryModelVersionsOperations,
68+
SchedulesOperations,
69+
ServerlessEndpointsOperations,
70+
UsagesOperations,
71+
VirtualMachineSizesOperations,
72+
WorkspaceConnectionsOperations,
73+
WorkspaceFeaturesOperations,
74+
WorkspacesOperations,
75+
)
2076

2177
if TYPE_CHECKING:
2278
# pylint: disable=unused-import,ungrouped-imports

‎sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/__init__.py‎

Lines changed: 31 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -6,61 +6,61 @@
66
# Changes may cause incorrect behavior and will be lost if the code is regenerated.
77
# --------------------------------------------------------------------------
88

9-
from ._usages_operations import UsagesOperations
10-
from ._virtual_machine_sizes_operations import VirtualMachineSizesOperations
11-
from ._quotas_operations import QuotasOperations
12-
from ._compute_operations import ComputeOperations
13-
from ._registries_operations import RegistriesOperations
14-
from ._workspace_features_operations import WorkspaceFeaturesOperations
15-
from ._capacity_reservation_groups_operations import CapacityReservationGroupsOperations
16-
from ._registry_code_containers_operations import RegistryCodeContainersOperations
17-
from ._registry_code_versions_operations import RegistryCodeVersionsOperations
18-
from ._registry_component_containers_operations import RegistryComponentContainersOperations
19-
from ._registry_component_versions_operations import RegistryComponentVersionsOperations
20-
from ._registry_data_containers_operations import RegistryDataContainersOperations
21-
from ._registry_data_versions_operations import RegistryDataVersionsOperations
22-
from ._registry_data_references_operations import RegistryDataReferencesOperations
23-
from ._registry_environment_containers_operations import RegistryEnvironmentContainersOperations
24-
from ._registry_environment_versions_operations import RegistryEnvironmentVersionsOperations
25-
from ._marketplace_subscriptions_operations import MarketplaceSubscriptionsOperations
26-
from ._registry_model_containers_operations import RegistryModelContainersOperations
27-
from ._registry_model_versions_operations import RegistryModelVersionsOperations
28-
from ._batch_endpoints_operations import BatchEndpointsOperations
299
from ._batch_deployments_operations import BatchDeploymentsOperations
10+
from ._batch_endpoints_operations import BatchEndpointsOperations
11+
from ._capacity_reservation_groups_operations import CapacityReservationGroupsOperations
3012
from ._code_containers_operations import CodeContainersOperations
3113
from ._code_versions_operations import CodeVersionsOperations
3214
from ._component_containers_operations import ComponentContainersOperations
3315
from ._component_versions_operations import ComponentVersionsOperations
16+
from ._compute_operations import ComputeOperations
3417
from ._data_containers_operations import DataContainersOperations
3518
from ._data_versions_operations import DataVersionsOperations
3619
from ._datastores_operations import DatastoresOperations
20+
from ._endpoint_deployment_operations import EndpointDeploymentOperations
21+
from ._endpoint_operations import EndpointOperations
3722
from ._environment_containers_operations import EnvironmentContainersOperations
3823
from ._environment_versions_operations import EnvironmentVersionsOperations
39-
from ._featureset_containers_operations import FeaturesetContainersOperations
4024
from ._features_operations import FeaturesOperations
25+
from ._featureset_containers_operations import FeaturesetContainersOperations
4126
from ._featureset_versions_operations import FeaturesetVersionsOperations
4227
from ._featurestore_entity_containers_operations import FeaturestoreEntityContainersOperations
4328
from ._featurestore_entity_versions_operations import FeaturestoreEntityVersionsOperations
44-
from ._inference_pools_operations import InferencePoolsOperations
4529
from ._inference_endpoints_operations import InferenceEndpointsOperations
4630
from ._inference_groups_operations import InferenceGroupsOperations
31+
from ._inference_pools_operations import InferencePoolsOperations
4732
from ._jobs_operations import JobsOperations
4833
from ._labeling_jobs_operations import LabelingJobsOperations
34+
from ._managed_network_provisions_operations import ManagedNetworkProvisionsOperations
35+
from ._managed_network_settings_rule_operations import ManagedNetworkSettingsRuleOperations
36+
from ._marketplace_subscriptions_operations import MarketplaceSubscriptionsOperations
4937
from ._model_containers_operations import ModelContainersOperations
5038
from ._model_versions_operations import ModelVersionsOperations
51-
from ._online_endpoints_operations import OnlineEndpointsOperations
5239
from ._online_deployments_operations import OnlineDeploymentsOperations
53-
from ._schedules_operations import SchedulesOperations
54-
from ._serverless_endpoints_operations import ServerlessEndpointsOperations
40+
from ._online_endpoints_operations import OnlineEndpointsOperations
5541
from ._operations import Operations
56-
from ._workspaces_operations import WorkspacesOperations
57-
from ._workspace_connections_operations import WorkspaceConnectionsOperations
58-
from ._endpoint_deployment_operations import EndpointDeploymentOperations
59-
from ._endpoint_operations import EndpointOperations
60-
from ._managed_network_settings_rule_operations import ManagedNetworkSettingsRuleOperations
6142
from ._private_endpoint_connections_operations import PrivateEndpointConnectionsOperations
6243
from ._private_link_resources_operations import PrivateLinkResourcesOperations
63-
from ._managed_network_provisions_operations import ManagedNetworkProvisionsOperations
44+
from ._quotas_operations import QuotasOperations
45+
from ._registries_operations import RegistriesOperations
46+
from ._registry_code_containers_operations import RegistryCodeContainersOperations
47+
from ._registry_code_versions_operations import RegistryCodeVersionsOperations
48+
from ._registry_component_containers_operations import RegistryComponentContainersOperations
49+
from ._registry_component_versions_operations import RegistryComponentVersionsOperations
50+
from ._registry_data_containers_operations import RegistryDataContainersOperations
51+
from ._registry_data_references_operations import RegistryDataReferencesOperations
52+
from ._registry_data_versions_operations import RegistryDataVersionsOperations
53+
from ._registry_environment_containers_operations import RegistryEnvironmentContainersOperations
54+
from ._registry_environment_versions_operations import RegistryEnvironmentVersionsOperations
55+
from ._registry_model_containers_operations import RegistryModelContainersOperations
56+
from ._registry_model_versions_operations import RegistryModelVersionsOperations
57+
from ._schedules_operations import SchedulesOperations
58+
from ._serverless_endpoints_operations import ServerlessEndpointsOperations
59+
from ._usages_operations import UsagesOperations
60+
from ._virtual_machine_sizes_operations import VirtualMachineSizesOperations
61+
from ._workspace_connections_operations import WorkspaceConnectionsOperations
62+
from ._workspace_features_operations import WorkspaceFeaturesOperations
63+
from ._workspaces_operations import WorkspacesOperations
6464

6565
__all__ = [
6666
'UsagesOperations',

‎sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_code_versions_operations.py‎

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,13 @@
99
from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar, Union
1010

1111
from azure.core.async_paging import AsyncItemPaged, AsyncList
12-
from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
12+
from azure.core.exceptions import (
13+
ClientAuthenticationError,
14+
HttpResponseError,
15+
ResourceExistsError,
16+
ResourceNotFoundError,
17+
map_error,
18+
)
1319
from azure.core.pipeline import PipelineResponse
1420
from azure.core.pipeline.transport import AsyncHttpResponse
1521
from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
@@ -21,7 +27,15 @@
2127

2228
from ... import models as _models
2329
from ..._vendor import _convert_request
24-
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
30+
from ...operations._code_versions_operations import (
31+
build_create_or_get_start_pending_upload_request,
32+
build_create_or_update_request,
33+
build_delete_request,
34+
build_get_request,
35+
build_list_request,
36+
build_publish_request_initial,
37+
)
38+
2539
T = TypeVar('T')
2640
ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
2741

‎sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_01_01_preview/aio/operations/_component_versions_operations.py‎

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,13 @@
99
from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar, Union
1010

1111
from azure.core.async_paging import AsyncItemPaged, AsyncList
12-
from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
12+
from azure.core.exceptions import (
13+
ClientAuthenticationError,
14+
HttpResponseError,
15+
ResourceExistsError,
16+
ResourceNotFoundError,
17+
map_error,
18+
)
1319
from azure.core.pipeline import PipelineResponse
1420
from azure.core.pipeline.transport import AsyncHttpResponse
1521
from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
@@ -21,7 +27,14 @@
2127

2228
from ... import models as _models
2329
from ..._vendor import _convert_request
24-
from ...operations._component_versions_operations import build_create_or_update_request, build_delete_request, build_get_request, build_list_request, build_publish_request_initial
30+
from ...operations._component_versions_operations import (
31+
build_create_or_update_request,
32+
build_delete_request,
33+
build_get_request,
34+
build_list_request,
35+
build_publish_request_initial,
36+
)
37+
2538
T = TypeVar('T')
2639
ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
2740

0 commit comments

Comments
 (0)