Skip to content

Commit 7d17d9b

Browse files
anyalee0221anyalee0221
authored
Update feature store related entities to use the GA version (#32541)
* Update feature store related entities to use GA version * Update the backfill job to use the 08-01 version API * Revert the change for backfill job, still use the 10-01 version API * Revert the change for feature store, it will still use the 04-01-preview version entity
1 parent bb81888 commit 7d17d9b

11 files changed

+18
-20
lines changed

‎sdk/ml/azure-ai-ml/azure/ai/ml/_utils/_feature_store_utils.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -12,14 +12,14 @@
1212
import yaml
1313

1414
from .._artifacts._artifact_utilities import get_datastore_info, get_storage_client
15-
from .._restclient.v2023_04_01_preview.operations import ( # pylint: disable = unused-import
15+
from .._restclient.v2023_10_01.operations import ( # pylint: disable = unused-import
1616
FeaturesetContainersOperations,
1717
FeaturesetVersionsOperations,
1818
FeaturestoreEntityContainersOperations,
1919
FeaturestoreEntityVersionsOperations,
2020
)
2121
from ..constants._common import DefaultOpenEncoding
22-
from ..exceptions import ValidationException, ErrorTarget, ErrorCategory, ValidationErrorType
22+
from ..exceptions import ErrorCategory, ErrorTarget, ValidationErrorType, ValidationException
2323
from ..operations._datastore_operations import DatastoreOperations
2424
from ._storage_utils import AzureMLDatastorePathUri
2525
from .utils import load_yaml

‎sdk/ml/azure-ai-ml/azure/ai/ml/entities/_assets/_artifacts/feature_set.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from pathlib import Path
99
from typing import IO, AnyStr, Dict, List, Optional, Union
1010

11-
from azure.ai.ml._restclient.v2023_02_01_preview.models import (
11+
from azure.ai.ml._restclient.v2023_10_01.models import (
1212
FeaturesetContainer,
1313
FeaturesetContainerProperties,
1414
FeaturesetVersion,

‎sdk/ml/azure-ai-ml/azure/ai/ml/entities/_feature_set/feature.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66

77
from typing import Dict, Optional
88

9-
from azure.ai.ml._restclient.v2023_04_01_preview.models import Feature as RestFeature
10-
from azure.ai.ml._restclient.v2023_04_01_preview.models import FeatureProperties
9+
from azure.ai.ml._restclient.v2023_10_01.models import Feature as RestFeature
10+
from azure.ai.ml._restclient.v2023_10_01.models import FeatureProperties
1111
from azure.ai.ml._utils._experimental import experimental
1212
from azure.ai.ml.entities._feature_store_entity.data_column_type import DataColumnType
1313
from azure.ai.ml.entities._mixins import RestTranslatableMixin

‎sdk/ml/azure-ai-ml/azure/ai/ml/entities/_feature_set/feature_set_backfill_metadata.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,11 @@
44

55
from typing import Optional
66

7-
from azure.ai.ml._restclient.v2023_08_01_preview.models import (
7+
from azure.ai.ml._restclient.v2023_10_01.models import (
88
FeaturesetVersionBackfillResponse as RestFeaturesetVersionBackfillResponse,
99
)
10-
from azure.ai.ml.entities._mixins import RestTranslatableMixin
1110
from azure.ai.ml._utils._experimental import experimental
11+
from azure.ai.ml.entities._mixins import RestTranslatableMixin
1212

1313

1414
@experimental

‎sdk/ml/azure-ai-ml/azure/ai/ml/entities/_feature_set/feature_set_backfill_request.py

+4-6
Original file line numberDiff line numberDiff line change
@@ -6,16 +6,14 @@
66

77
from os import PathLike
88
from pathlib import Path
9-
from typing import Dict, Optional, Union, List
9+
from typing import Dict, List, Optional, Union
1010

11-
from azure.ai.ml._restclient.v2023_08_01_preview.models import FeatureWindow
12-
from azure.ai.ml.entities._mixins import RestTranslatableMixin
13-
from azure.ai.ml.entities._feature_set.materialization_compute_resource import (
14-
MaterializationComputeResource,
15-
)
11+
from azure.ai.ml._restclient.v2023_10_01.models import FeatureWindow
1612
from azure.ai.ml._schema._feature_set.feature_set_backfill_schema import FeatureSetBackfillSchema
1713
from azure.ai.ml._utils._experimental import experimental
1814
from azure.ai.ml.constants._common import BASE_PATH_CONTEXT_KEY, PARAMS_OVERRIDE_KEY
15+
from azure.ai.ml.entities._feature_set.materialization_compute_resource import MaterializationComputeResource
16+
from azure.ai.ml.entities._mixins import RestTranslatableMixin
1917
from azure.ai.ml.entities._util import load_from_dict
2018

2119

‎sdk/ml/azure-ai-ml/azure/ai/ml/entities/_feature_set/feature_set_materialization_metadata.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from datetime import datetime, timedelta
66
from typing import Dict, Optional
77

8-
from azure.ai.ml._restclient.v2023_08_01_preview.models import JobBase as RestJobBase
8+
from azure.ai.ml._restclient.v2023_10_01.models import JobBase as RestJobBase
99
from azure.ai.ml._utils._experimental import experimental
1010
from azure.ai.ml.entities._mixins import RestTranslatableMixin
1111
from azure.ai.ml.entities._system_data import SystemData

‎sdk/ml/azure-ai-ml/azure/ai/ml/entities/_feature_set/feature_set_specification.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
from typing import Optional
66

7-
from azure.ai.ml._restclient.v2023_04_01_preview.models import FeaturesetSpecification as RestFeaturesetSpecification
7+
from azure.ai.ml._restclient.v2023_10_01.models import FeaturesetSpecification as RestFeaturesetSpecification
88
from azure.ai.ml._utils._experimental import experimental
99
from azure.ai.ml.entities._mixins import RestTranslatableMixin
1010

‎sdk/ml/azure-ai-ml/azure/ai/ml/entities/_feature_set/materialization_compute_resource.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Copyright (c) Microsoft Corporation. All rights reserved.
33
# ---------------------------------------------------------
44

5-
from azure.ai.ml._restclient.v2023_04_01_preview.models import (
5+
from azure.ai.ml._restclient.v2023_10_01.models import (
66
MaterializationComputeResource as RestMaterializationComputeResource,
77
)
88
from azure.ai.ml._utils._experimental import experimental

‎sdk/ml/azure-ai-ml/azure/ai/ml/entities/_feature_set/materialization_settings.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@
44

55
from typing import Dict, Optional
66

7-
from azure.ai.ml._restclient.v2023_04_01_preview.models import MaterializationSettings as RestMaterializationSettings
8-
from azure.ai.ml._restclient.v2023_04_01_preview.models import MaterializationStoreType
7+
from azure.ai.ml._restclient.v2023_10_01.models import MaterializationSettings as RestMaterializationSettings
8+
from azure.ai.ml._restclient.v2023_10_01.models import MaterializationStoreType
99
from azure.ai.ml._utils._experimental import experimental
1010
from azure.ai.ml.entities._feature_set.materialization_compute_resource import MaterializationComputeResource
1111
from azure.ai.ml.entities._mixins import RestTranslatableMixin

‎sdk/ml/azure-ai-ml/azure/ai/ml/entities/_feature_store_entity/data_column.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
from typing import Dict, Optional, Union
88

9-
from azure.ai.ml._restclient.v2023_02_01_preview.models import FeatureDataType, IndexColumn
9+
from azure.ai.ml._restclient.v2023_10_01.models import FeatureDataType, IndexColumn
1010
from azure.ai.ml._utils._experimental import experimental
1111
from azure.ai.ml.entities._mixins import RestTranslatableMixin
1212
from azure.ai.ml.exceptions import ErrorCategory, ErrorTarget, ValidationErrorType, ValidationException

‎sdk/ml/azure-ai-ml/azure/ai/ml/entities/_feature_store_entity/feature_store_entity.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from pathlib import Path
99
from typing import Dict, List, Optional, Union
1010

11-
from azure.ai.ml._restclient.v2023_04_01_preview.models import (
11+
from azure.ai.ml._restclient.v2023_10_01.models import (
1212
FeaturestoreEntityContainer,
1313
FeaturestoreEntityContainerProperties,
1414
FeaturestoreEntityVersion,

0 commit comments

Comments
 (0)