Skip to content

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

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
TonyJ1 merged 19 commits into from
Mar 3, 2023
Merged

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

Show file tree
Hide file tree
Changes from 5 commits
Commits
Show all changes
19 commits
Select commit Hold shift + click to select a range
d1faa05
[ML] Migrate from 2022_12_01_preview API to 2023_02_01_preview API
TonyJ1 Feb 16, 2023
b2a33da
Fix ml_client
TonyJ1 Feb 16, 2023
e113d69
Hardcode MFE url to master for testing the gates
TonyJ1 Feb 16, 2023
f3f0aed
Merge main
TonyJ1 Feb 24, 2023
e5f8d45
TEMP change to sanitize MFE url
TonyJ1 Feb 25, 2023
7d53ebe
Merge main
TonyJ1 Feb 28, 2023
51e31ab
Merge branch 'main' of https://github.com/TonyJ1/azure-sdk-for-python…
TonyJ1 Mar 1, 2023
6f0d403
Automl
TonyJ1 Mar 2, 2023
55229d7
MLClient and Workspace
TonyJ1 Mar 2, 2023
7697ada
Merge main
TonyJ1 Mar 2, 2023
38cf026
Workspace and Networking still depend on 2022-12
TonyJ1 Mar 2, 2023
0475806
ComputeRuntimeDTO and FeatureStoreSettings depends on v2022_12_01_pre…
TonyJ1 Mar 2, 2023
5bf25db
Merge branch 'main' of https://github.com/TonyJ1/azure-sdk-for-python…
TonyJ1 Mar 2, 2023
3a7be00
Move remaining workspace files to use 2022-12 API
TonyJ1 Mar 2, 2023
20e54be
Move workspace_keys files to use 2022-12 API
TonyJ1 Mar 2, 2023
4e62db4
Move test_ws_operations to use 2022-12 API
TonyJ1 Mar 2, 2023
7ce6321
Remove SAS URI Santizer
TonyJ1 Mar 2, 2023
04d1a50
Merge branch 'main' of https://github.com/TonyJ1/azure-sdk-for-python…
TonyJ1 Mar 2, 2023
4fd6d80
Remove hardcodes to use master mfe endpoint
TonyJ1 Mar 3, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions sdk/ml/azure-ai-ml/azure/ai/ml/_internal/entities/command.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,8 @@
from azure.ai.ml._internal._schema.component import NodeType
from azure.ai.ml._internal.entities.component import InternalComponent
from azure.ai.ml._internal.entities.node import InternalBaseNode
from azure.ai.ml._restclient.v2022_12_01_preview.models import CommandJobLimits as RestCommandJobLimits
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
from azure.ai.ml._restclient.v2023_02_01_preview.models import CommandJobLimits as RestCommandJobLimits
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
from azure.ai.ml._schema import PathAwareSchema
from azure.ai.ml._schema.core.fields import DistributionField
from azure.ai.ml.entities import CommandJobLimits, JobResourceConfiguration
Expand Down
7 changes: 5 additions & 2 deletions sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -285,7 +285,10 @@ def __init__(
self._service_client_02_2023_preview = ServiceClient022023Preview(
credential=self._credential,
subscription_id=self._operation_scope._subscription_id,
base_url=base_url,
# base_url=base_url,
Comment thread
TonyJ1 marked this conversation as resolved.
Outdated
# TODO: Below hardcode is just for testing the gates
# TODO: REMOVE before the merge
Comment thread
TonyJ1 marked this conversation as resolved.
Outdated
base_url="https://master.api.azureml-test.ms/mferp/managementfrontend",
**kwargs,
)

Expand Down Expand Up @@ -426,7 +429,7 @@ def __init__(
self._jobs = JobOperations(
self._operation_scope,
self._operation_config,
self._service_client_12_2022_preview,
self._service_client_02_2023_preview,
self._operation_container,
self._credential,
_service_client_kwargs=kwargs,
Expand Down
8 changes: 4 additions & 4 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/command.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,10 +13,10 @@

from marshmallow import INCLUDE, Schema

from azure.ai.ml._restclient.v2022_12_01_preview.models import CommandJob as RestCommandJob
from azure.ai.ml._restclient.v2022_12_01_preview.models import CommandJobLimits as RestCommandJobLimits
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
from azure.ai.ml._restclient.v2023_02_01_preview.models import CommandJob as RestCommandJob
from azure.ai.ml._restclient.v2023_02_01_preview.models import CommandJobLimits as RestCommandJobLimits
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
from azure.ai.ml._schema.core.fields import NestedField, UnionField
from azure.ai.ml._schema.job.command_job import CommandJobSchema
from azure.ai.ml._schema.job.identity import AMLTokenIdentitySchema, ManagedIdentitySchema, UserIdentitySchema
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/parallel.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@

from marshmallow import Schema

from azure.ai.ml._restclient.v2022_12_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobResourceConfiguration as RestJobResourceConfiguration
from azure.ai.ml.constants._common import ARM_ID_PREFIX
from azure.ai.ml.constants._component import NodeType
from azure.ai.ml.entities._component.component import Component
Expand Down
10 changes: 5 additions & 5 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/spark.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,11 +12,11 @@

from marshmallow import INCLUDE, Schema

from azure.ai.ml._restclient.v2022_12_01_preview.models import IdentityConfiguration
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase as JobBaseData
from azure.ai.ml._restclient.v2022_12_01_preview.models import SparkJob as RestSparkJob
from azure.ai.ml._restclient.v2022_12_01_preview.models import SparkJobEntry as RestSparkJobEntry
from azure.ai.ml._restclient.v2022_12_01_preview.models import (
from azure.ai.ml._restclient.v2023_02_01_preview.models import IdentityConfiguration
Comment thread
TonyJ1 marked this conversation as resolved.
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase as JobBaseData
from azure.ai.ml._restclient.v2023_02_01_preview.models import SparkJob as RestSparkJob
from azure.ai.ml._restclient.v2023_02_01_preview.models import SparkJobEntry as RestSparkJobEntry
from azure.ai.ml._restclient.v2023_02_01_preview.models import (
SparkResourceConfiguration as RestSparkResourceConfiguration,
)
from azure.ai.ml._schema.job.identity import AMLTokenIdentitySchema, ManagedIdentitySchema, UserIdentitySchema
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/spark_func.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import os
from typing import Callable, Dict, List, Optional, Tuple, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import AmlToken, ManagedIdentity, UserIdentity
from azure.ai.ml._restclient.v2023_02_01_preview.models import AmlToken, ManagedIdentity, UserIdentity
from azure.ai.ml.constants._common import AssetTypes
from azure.ai.ml.constants._component import ComponentSource
from azure.ai.ml.entities import Environment
Expand Down
12 changes: 6 additions & 6 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_credentials.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,12 +43,12 @@
ServicePrincipalDatastoreSecrets as RestServicePrincipalDatastoreSecrets,
)
from azure.ai.ml._restclient.v2022_05_01.models import UserAssignedIdentity as RestUserAssignedIdentityConfiguration
from azure.ai.ml._restclient.v2022_12_01_preview.models import AmlToken as RestAmlToken
from azure.ai.ml._restclient.v2022_12_01_preview.models import IdentityConfiguration as RestJobIdentityConfiguration
from azure.ai.ml._restclient.v2022_12_01_preview.models import IdentityConfigurationType
from azure.ai.ml._restclient.v2022_12_01_preview.models import ManagedIdentity as RestJobManagedIdentity
from azure.ai.ml._restclient.v2022_12_01_preview.models import ManagedServiceIdentity as RestRegistryManagedIdentity
from azure.ai.ml._restclient.v2022_12_01_preview.models import UserIdentity as RestUserIdentity
from azure.ai.ml._restclient.v2023_02_01_preview.models import AmlToken as RestAmlToken
from azure.ai.ml._restclient.v2023_02_01_preview.models import IdentityConfiguration as RestJobIdentityConfiguration
from azure.ai.ml._restclient.v2023_02_01_preview.models import IdentityConfigurationType
from azure.ai.ml._restclient.v2023_02_01_preview.models import ManagedIdentity as RestJobManagedIdentity
from azure.ai.ml._restclient.v2023_02_01_preview.models import ManagedServiceIdentity as RestRegistryManagedIdentity
from azure.ai.ml._restclient.v2023_02_01_preview.models import UserIdentity as RestUserIdentity
from azure.ai.ml._utils.utils import camel_to_snake, snake_to_pascal
from azure.ai.ml.constants._common import CommonYamlFields, IdentityType
from azure.ai.ml.entities._mixins import DictMixin, RestTranslatableMixin, YamlTranslatableMixin
Expand Down
36 changes: 18 additions & 18 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/_input_output_helpers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,24 +6,24 @@
import re
from typing import Any, Dict, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import CustomModelJobInput as RestCustomModelJobInput
from azure.ai.ml._restclient.v2022_12_01_preview.models import CustomModelJobOutput as RestCustomModelJobOutput
from azure.ai.ml._restclient.v2022_12_01_preview.models import InputDeliveryMode
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobInput as RestJobInput
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobInputType
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobOutput as RestJobOutput
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobOutputType, LiteralJobInput
from azure.ai.ml._restclient.v2022_12_01_preview.models import MLFlowModelJobInput as RestMLFlowModelJobInput
from azure.ai.ml._restclient.v2022_12_01_preview.models import MLFlowModelJobOutput as RestMLFlowModelJobOutput
from azure.ai.ml._restclient.v2022_12_01_preview.models import MLTableJobInput as RestMLTableJobInput
from azure.ai.ml._restclient.v2022_12_01_preview.models import MLTableJobOutput as RestMLTableJobOutput
from azure.ai.ml._restclient.v2022_12_01_preview.models import OutputDeliveryMode
from azure.ai.ml._restclient.v2022_12_01_preview.models import TritonModelJobInput as RestTritonModelJobInput
from azure.ai.ml._restclient.v2022_12_01_preview.models import TritonModelJobOutput as RestTritonModelJobOutput
from azure.ai.ml._restclient.v2022_12_01_preview.models import UriFileJobInput as RestUriFileJobInput
from azure.ai.ml._restclient.v2022_12_01_preview.models import UriFileJobOutput as RestUriFileJobOutput
from azure.ai.ml._restclient.v2022_12_01_preview.models import UriFolderJobInput as RestUriFolderJobInput
from azure.ai.ml._restclient.v2022_12_01_preview.models import UriFolderJobOutput as RestUriFolderJobOutput
from azure.ai.ml._restclient.v2023_02_01_preview.models import CustomModelJobInput as RestCustomModelJobInput
from azure.ai.ml._restclient.v2023_02_01_preview.models import CustomModelJobOutput as RestCustomModelJobOutput
from azure.ai.ml._restclient.v2023_02_01_preview.models import InputDeliveryMode
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobInput as RestJobInput
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobInputType
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobOutput as RestJobOutput
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobOutputType, LiteralJobInput
from azure.ai.ml._restclient.v2023_02_01_preview.models import MLFlowModelJobInput as RestMLFlowModelJobInput
from azure.ai.ml._restclient.v2023_02_01_preview.models import MLFlowModelJobOutput as RestMLFlowModelJobOutput
from azure.ai.ml._restclient.v2023_02_01_preview.models import MLTableJobInput as RestMLTableJobInput
from azure.ai.ml._restclient.v2023_02_01_preview.models import MLTableJobOutput as RestMLTableJobOutput
from azure.ai.ml._restclient.v2023_02_01_preview.models import OutputDeliveryMode
from azure.ai.ml._restclient.v2023_02_01_preview.models import TritonModelJobInput as RestTritonModelJobInput
from azure.ai.ml._restclient.v2023_02_01_preview.models import TritonModelJobOutput as RestTritonModelJobOutput
from azure.ai.ml._restclient.v2023_02_01_preview.models import UriFileJobInput as RestUriFileJobInput
from azure.ai.ml._restclient.v2023_02_01_preview.models import UriFileJobOutput as RestUriFileJobOutput
from azure.ai.ml._restclient.v2023_02_01_preview.models import UriFolderJobInput as RestUriFolderJobInput
from azure.ai.ml._restclient.v2023_02_01_preview.models import UriFolderJobOutput as RestUriFolderJobOutput
from azure.ai.ml._utils.utils import is_data_binding_expression
from azure.ai.ml.constants import AssetTypes, InputOutputModes, JobType
from azure.ai.ml.constants._component import IOConstants
Expand Down
4 changes: 2 additions & 2 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/command_job.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,8 @@
from pathlib import Path
from typing import Dict, Optional, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import CommandJob as RestCommandJob
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase
from azure.ai.ml._restclient.v2023_02_01_preview.models import CommandJob as RestCommandJob
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase
from azure.ai.ml._schema.job.command_job import CommandJobSchema
from azure.ai.ml._utils.utils import map_single_brackets_and_warn
from azure.ai.ml.constants import JobType
Expand Down
10 changes: 5 additions & 5 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/distribution.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,13 +6,13 @@

from typing import Dict, Optional, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import (
from azure.ai.ml._restclient.v2023_02_01_preview.models import (
DistributionConfiguration as RestDistributionConfiguration,
)
from azure.ai.ml._restclient.v2022_12_01_preview.models import DistributionType as RestDistributionType
from azure.ai.ml._restclient.v2022_12_01_preview.models import Mpi as RestMpi
from azure.ai.ml._restclient.v2022_12_01_preview.models import PyTorch as RestPyTorch
from azure.ai.ml._restclient.v2022_12_01_preview.models import TensorFlow as RestTensorFlow
from azure.ai.ml._restclient.v2023_02_01_preview.models import DistributionType as RestDistributionType
from azure.ai.ml._restclient.v2023_02_01_preview.models import Mpi as RestMpi
from azure.ai.ml._restclient.v2023_02_01_preview.models import PyTorch as RestPyTorch
from azure.ai.ml._restclient.v2023_02_01_preview.models import TensorFlow as RestTensorFlow
from azure.ai.ml.constants import DistributionType
from azure.ai.ml.entities._mixins import RestTranslatableMixin

Expand Down
4 changes: 2 additions & 2 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/job.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,8 @@
from typing import IO, AnyStr, Dict, Optional, Type, Union

from azure.ai.ml._restclient.runhistory.models import Run
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase, JobService
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobType as RestJobType
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase, JobService
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobType as RestJobType
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
Expand Down
4 changes: 2 additions & 2 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/pipeline/_io/mixin.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,8 @@
import copy
from typing import Dict, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import JobInput as RestJobInput
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobOutput as RestJobOutput
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobInput as RestJobInput
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobOutput as RestJobOutput
from azure.ai.ml.constants._component import ComponentJobConstants
from azure.ai.ml.entities._inputs_outputs import GroupInput, Input, Output
from azure.ai.ml.entities._util import copy_output_setting
Expand Down
8 changes: 4 additions & 4 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/pipeline/_pipeline_job_helpers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,10 @@
import re
from typing import Dict, List, Tuple, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import InputDeliveryMode
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobInput as RestJobInput
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobOutput as RestJobOutput
from azure.ai.ml._restclient.v2022_12_01_preview.models import Mpi, PyTorch, TensorFlow
from azure.ai.ml._restclient.v2023_02_01_preview.models import InputDeliveryMode
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobInput as RestJobInput
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobOutput as RestJobOutput
from azure.ai.ml._restclient.v2023_02_01_preview.models import Mpi, PyTorch, TensorFlow
from azure.ai.ml.constants._component import ComponentJobConstants
from azure.ai.ml.entities._inputs_outputs import Input, Output
from azure.ai.ml.entities._job._input_output_helpers import (
Expand Down
4 changes: 2 additions & 2 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/pipeline/pipeline_job.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,8 @@
from pathlib import Path
from typing import Dict, List, Optional, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase
from azure.ai.ml._restclient.v2022_12_01_preview.models import PipelineJob as RestPipelineJob
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase
from azure.ai.ml._restclient.v2023_02_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
Expand Down
4 changes: 2 additions & 2 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/spark_job.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,8 @@

from marshmallow import INCLUDE

from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase
from azure.ai.ml._restclient.v2022_12_01_preview.models import SparkJob as RestSparkJob
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase
from azure.ai.ml._restclient.v2023_02_01_preview.models import SparkJob as RestSparkJob
from azure.ai.ml._schema.job.identity import AMLTokenIdentitySchema, ManagedIdentitySchema, UserIdentitySchema
from azure.ai.ml._schema.job.parameterized_spark import CONF_KEY_MAP, SparkConfSchema
from azure.ai.ml._schema.job.spark_job import SparkJobSchema
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/spark_resource_configuration.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

from typing import Optional

from azure.ai.ml._restclient.v2022_12_01_preview.models import (
from azure.ai.ml._restclient.v2023_02_01_preview.models import (
SparkResourceConfiguration as RestSparkResourceConfiguration,
)
from azure.ai.ml.entities._mixins import DictMixin, RestTranslatableMixin
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/to_rest_functions.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
from functools import singledispatch
from pathlib import Path

from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase as JobBaseData
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase as JobBaseData
from azure.ai.ml.constants._common import DEFAULT_EXPERIMENT_NAME
from azure.ai.ml.entities._builders.command import Command
from azure.ai.ml.entities._builders.pipeline import Pipeline
Expand Down
10 changes: 5 additions & 5 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_schedule/schedule.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,11 +7,11 @@
from pathlib import Path
from typing import IO, AnyStr, Dict, Optional, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase as RestJobBase
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobScheduleAction
from azure.ai.ml._restclient.v2022_12_01_preview.models import PipelineJob as RestPipelineJob
from azure.ai.ml._restclient.v2022_12_01_preview.models import Schedule as RestSchedule
from azure.ai.ml._restclient.v2022_12_01_preview.models import ScheduleProperties
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase as RestJobBase
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobScheduleAction
from azure.ai.ml._restclient.v2023_02_01_preview.models import PipelineJob as RestPipelineJob
from azure.ai.ml._restclient.v2023_02_01_preview.models import Schedule as RestSchedule
from azure.ai.ml._restclient.v2023_02_01_preview.models import ScheduleProperties
from azure.ai.ml._schema.schedule.schedule import ScheduleSchema
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
Expand Down
10 changes: 5 additions & 5 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_schedule/trigger.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,11 +7,11 @@
from datetime import datetime
from typing import List, Optional, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import CronTrigger as RestCronTrigger
from azure.ai.ml._restclient.v2022_12_01_preview.models import RecurrenceSchedule as RestRecurrencePattern
from azure.ai.ml._restclient.v2022_12_01_preview.models import RecurrenceTrigger as RestRecurrenceTrigger
from azure.ai.ml._restclient.v2022_12_01_preview.models import TriggerBase as RestTriggerBase
from azure.ai.ml._restclient.v2022_12_01_preview.models import TriggerType as RestTriggerType
from azure.ai.ml._restclient.v2023_02_01_preview.models import CronTrigger as RestCronTrigger
from azure.ai.ml._restclient.v2023_02_01_preview.models import RecurrenceSchedule as RestRecurrencePattern
from azure.ai.ml._restclient.v2023_02_01_preview.models import RecurrenceTrigger as RestRecurrenceTrigger
from azure.ai.ml._restclient.v2023_02_01_preview.models import TriggerBase as RestTriggerBase
from azure.ai.ml._restclient.v2023_02_01_preview.models import TriggerType as RestTriggerType
from azure.ai.ml._utils.utils import camel_to_snake, snake_to_camel
from azure.ai.ml.constants import TimeZone
from azure.ai.ml.entities._mixins import RestTranslatableMixin
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/entities/_util.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
from marshmallow.exceptions import ValidationError

from azure.ai.ml._restclient.v2022_02_01_preview.models import JobInputType as JobInputType02
from azure.ai.ml._restclient.v2022_12_01_preview.models import JobInputType as JobInputType10
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobInputType as JobInputType10
from azure.ai.ml._schema._datastore import (
AzureBlobSchema,
AzureDataLakeGen1Schema,
Expand Down
4 changes: 2 additions & 2 deletions sdk/ml/azure-ai-ml/azure/ai/ml/operations/_compute_operations.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@

from typing import Any, Dict, Iterable, Optional

from azure.ai.ml._restclient.v2022_12_01_preview import AzureMachineLearningWorkspaces as ServiceClient122022Preview
from azure.ai.ml._restclient.v2023_02_01_preview import AzureMachineLearningWorkspaces as ServiceClient022023Preview
from azure.ai.ml._scope_dependent_operations import OperationConfig, OperationScope, _ScopeDependentOperations

# from azure.ai.ml._telemetry import ActivityType, monitor_with_activity
Expand All @@ -33,7 +33,7 @@ def __init__(
self,
operation_scope: OperationScope,
operation_config: OperationConfig,
service_client: ServiceClient122022Preview,
service_client: ServiceClient022023Preview,
**kwargs: Dict,
):
super(ComputeOperations, self).__init__(operation_scope, operation_config)
Expand Down
Loading