Skip to content
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

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

Merged
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 13 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
9 changes: 6 additions & 3 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 @@ -263,7 +263,7 @@ def __init__(
**kwargs,
)

self._rp_service_client = ServiceClient122022Preview(
self._rp_service_client = ServiceClient022023Preview(
TonyJ1 marked this conversation as resolved.
Show resolved Hide resolved
subscription_id=self._operation_scope._subscription_id,
credential=self._credential,
base_url=base_url,
Expand Down Expand Up @@ -324,7 +324,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,
TonyJ1 marked this conversation as resolved.
Show resolved Hide resolved
# TODO: Below hardcode is just for testing the gates
# TODO: REMOVE before the merge
TonyJ1 marked this conversation as resolved.
Show resolved Hide resolved
base_url="https://master.api.azureml-test.ms/mferp/managementfrontend",
**kwargs,
)

Expand Down Expand Up @@ -465,7 +468,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
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/automl_vertical.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
# Copyright (c) Microsoft Corporation. All rights reserved.
# ---------------------------------------------------------

from azure.ai.ml._restclient.v2022_12_01_preview.models import LogVerbosity
from azure.ai.ml._restclient.v2023_02_01_preview.models import LogVerbosity
from azure.ai.ml._schema.automl.automl_job import AutoMLJobSchema
from azure.ai.ml._schema.core.fields import NestedField, StringTransformedEnum, UnionField
from azure.ai.ml._schema.job.input_output_entry import MLTableInputSchema
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/featurization_settings.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
from marshmallow import fields as flds
from marshmallow import post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import BlockedTransformers
from azure.ai.ml._restclient.v2023_02_01_preview.models import BlockedTransformers
from azure.ai.ml._schema.core.fields import NestedField, StringTransformedEnum, UnionField
from azure.ai.ml._schema.core.schema import PatchedSchemaMeta
from azure.ai.ml._utils.utils import camel_to_snake
Expand Down
6 changes: 3 additions & 3 deletions sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/forecasting_settings.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,16 +6,16 @@

from marshmallow import fields, post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import FeatureLags as FeatureLagsMode
from azure.ai.ml._restclient.v2022_12_01_preview.models import (
from azure.ai.ml._restclient.v2023_02_01_preview.models import FeatureLags as FeatureLagsMode
from azure.ai.ml._restclient.v2023_02_01_preview.models import (
ForecastHorizonMode,
SeasonalityMode,
ShortSeriesHandlingConfiguration,
TargetAggregationFunction,
TargetLagsMode,
TargetRollingWindowSizeMode,
)
from azure.ai.ml._restclient.v2022_12_01_preview.models import UseStl as STLMode
from azure.ai.ml._restclient.v2023_02_01_preview.models import UseStl as STLMode
from azure.ai.ml._schema.core.fields import StringTransformedEnum, UnionField
from azure.ai.ml._schema.core.schema import PatchedSchemaMeta

Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_classification.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

from marshmallow import fields, post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import (
from azure.ai.ml._restclient.v2023_02_01_preview.models import (
ClassificationMultilabelPrimaryMetrics,
ClassificationPrimaryMetrics,
TaskType,
Expand Down
2 changes: 1 addition & 1 deletion ...zure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_model_distribution_settings.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@

from marshmallow import fields, post_dump, post_load, pre_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import (
from azure.ai.ml._restclient.v2023_02_01_preview.models import (
LearningRateScheduler,
ModelSize,
StochasticOptimizer,
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_model_settings.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@

from marshmallow import fields, post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import (
from azure.ai.ml._restclient.v2023_02_01_preview.models import (
LearningRateScheduler,
ModelSize,
StochasticOptimizer,
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/image_vertical/image_object_detection.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

from marshmallow import fields, post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import (
from azure.ai.ml._restclient.v2023_02_01_preview.models import (
InstanceSegmentationPrimaryMetrics,
ObjectDetectionPrimaryMetrics,
TaskType,
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/nlp_fixed_parameters.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@

from marshmallow import fields, post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import NlpLearningRateScheduler
from azure.ai.ml._restclient.v2023_02_01_preview.models import NlpLearningRateScheduler
from azure.ai.ml._schema.core.fields import StringTransformedEnum
from azure.ai.ml._schema.core.schema import PatchedSchemaMeta
from azure.ai.ml._utils.utils import camel_to_snake, snake_to_kebab
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/nlp_parameter_subspace.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@

from marshmallow import fields, post_dump, post_load, pre_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import NlpLearningRateScheduler
from azure.ai.ml._restclient.v2023_02_01_preview.models import NlpLearningRateScheduler
from azure.ai.ml._schema._sweep.search_space import (
ChoiceSchema,
NormalSchema,
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/text_classification.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

from marshmallow import post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import ClassificationPrimaryMetrics, TaskType
from azure.ai.ml._restclient.v2023_02_01_preview.models import ClassificationPrimaryMetrics, TaskType
from azure.ai.ml._schema.automl.nlp_vertical.nlp_vertical import NlpVerticalSchema
from azure.ai.ml._schema.core.fields import StringTransformedEnum, fields
from azure.ai.ml._utils.utils import camel_to_snake
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/text_classification_multilabel.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

from marshmallow import post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import ClassificationMultilabelPrimaryMetrics, TaskType
from azure.ai.ml._restclient.v2023_02_01_preview.models import ClassificationMultilabelPrimaryMetrics, TaskType
from azure.ai.ml._schema.automl.nlp_vertical.nlp_vertical import NlpVerticalSchema
from azure.ai.ml._schema.core.fields import StringTransformedEnum, fields
from azure.ai.ml._utils.utils import camel_to_snake
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/nlp_vertical/text_ner.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

from marshmallow import post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import ClassificationPrimaryMetrics, TaskType
from azure.ai.ml._restclient.v2023_02_01_preview.models import ClassificationPrimaryMetrics, TaskType
from azure.ai.ml._schema.automl.nlp_vertical.nlp_vertical import NlpVerticalSchema
from azure.ai.ml._schema.core.fields import StringTransformedEnum, fields
from azure.ai.ml._utils.utils import camel_to_snake
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/classification.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

from marshmallow import fields, post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import ClassificationPrimaryMetrics, TaskType
from azure.ai.ml._restclient.v2023_02_01_preview.models import ClassificationPrimaryMetrics, TaskType
from azure.ai.ml._schema.automl.table_vertical.table_vertical import AutoMLTableVerticalSchema
from azure.ai.ml._schema.automl.training_settings import ClassificationTrainingSettingsSchema
from azure.ai.ml._schema.core.fields import NestedField, StringTransformedEnum
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/forecasting.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

from marshmallow import post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import ForecastingPrimaryMetrics, TaskType
from azure.ai.ml._restclient.v2023_02_01_preview.models import ForecastingPrimaryMetrics, TaskType
from azure.ai.ml._schema.automl.forecasting_settings import ForecastingSettingsSchema
from azure.ai.ml._schema.automl.table_vertical.table_vertical import AutoMLTableVerticalSchema
from azure.ai.ml._schema.automl.training_settings import ForecastingTrainingSettingsSchema
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/regression.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

from marshmallow import post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import RegressionPrimaryMetrics, TaskType
from azure.ai.ml._restclient.v2023_02_01_preview.models import RegressionPrimaryMetrics, TaskType
from azure.ai.ml._schema.automl.table_vertical.table_vertical import AutoMLTableVerticalSchema
from azure.ai.ml._schema.automl.training_settings import RegressionTrainingSettingsSchema
from azure.ai.ml._schema.core.fields import NestedField, StringTransformedEnum
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/table_vertical/table_vertical.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

# pylint: disable=unused-argument,no-self-use

from azure.ai.ml._restclient.v2022_12_01_preview.models import NCrossValidationsMode
from azure.ai.ml._restclient.v2023_02_01_preview.models import NCrossValidationsMode
from azure.ai.ml._schema.automl.automl_vertical import AutoMLVerticalSchema
from azure.ai.ml._schema.automl.featurization_settings import TableFeaturizationSettingsSchema
from azure.ai.ml._schema.automl.table_vertical.table_vertical_limit_settings import AutoMLTableLimitsSchema
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/_schema/automl/training_settings.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@

from marshmallow import fields, post_load

from azure.ai.ml._restclient.v2022_12_01_preview.models import (
from azure.ai.ml._restclient.v2023_02_01_preview.models import (
ClassificationModels,
ForecastingModels,
RegressionModels,
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/automl/_automl_image.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 Optional, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import (
from azure.ai.ml._restclient.v2023_02_01_preview.models import (
ClassificationMultilabelPrimaryMetrics,
ClassificationPrimaryMetrics,
InstanceSegmentationPrimaryMetrics,
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
TonyJ1 marked this conversation as resolved.
Show resolved Hide 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
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/automl/automl_job.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from abc import ABC, abstractmethod
from typing import Any, Dict, Optional, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import JobBase, MLTableJobInput, ResourceConfiguration, TaskType
from azure.ai.ml._restclient.v2023_02_01_preview.models import JobBase, MLTableJobInput, ResourceConfiguration, TaskType
from azure.ai.ml._utils.utils import camel_to_snake
from azure.ai.ml.constants import JobType
from azure.ai.ml.constants._common import TYPE, AssetTypes
Expand Down
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/automl/image/automl_image.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from abc import ABC
from typing import Dict, Optional, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import LogVerbosity, SamplingAlgorithmType
from azure.ai.ml._restclient.v2023_02_01_preview.models import LogVerbosity, SamplingAlgorithmType
from azure.ai.ml._utils.utils import camel_to_snake
from azure.ai.ml.entities._inputs_outputs import Input
from azure.ai.ml.entities._job.automl.automl_vertical import AutoMLVertical
Expand Down
2 changes: 1 addition & 1 deletion ...ml/azure-ai-ml/azure/ai/ml/entities/_job/automl/image/automl_image_classification_base.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 Dict, List, Optional, Union

from azure.ai.ml._restclient.v2022_12_01_preview.models import LearningRateScheduler, StochasticOptimizer
from azure.ai.ml._restclient.v2023_02_01_preview.models import LearningRateScheduler, StochasticOptimizer
from azure.ai.ml._utils.utils import camel_to_snake
from azure.ai.ml.entities._job.automl.image.automl_image import AutoMLImage
from azure.ai.ml.entities._job.automl.image.image_classification_search_space import ImageClassificationSearchSpace
Expand Down
Loading