sagemaker-python-sdk/sagemaker-mlops/src/sagemaker/mlops/feature_store/__init__.py at 23371cae87dcd59f523887e2f4c5671dcf07ed83 · nargokul/sagemaker-python-sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# Licensed under the Apache License, Version 2.0
"""SageMaker FeatureStore V3 - powered by sagemaker-core."""

# Resources from core
from sagemaker.core.resources import FeatureGroup, FeatureMetadata

# Shapes from core (Pydantic - no to_dict() needed)
from sagemaker.core.shapes import (
    DataCatalogConfig,
    FeatureParameter,
    FeatureValue,
    Filter,
    OfflineStoreConfig,
    OnlineStoreConfig,
    OnlineStoreSecurityConfig,
    S3StorageConfig,
    SearchExpression,
    ThroughputConfig,
    TtlDuration,
)

# Enums (local - core uses strings)
from sagemaker.mlops.feature_store.inputs import (
    DeletionModeEnum,
    ExpirationTimeResponseEnum,
    FilterOperatorEnum,
    OnlineStoreStorageTypeEnum,
    ResourceEnum,
    SearchOperatorEnum,
    SortOrderEnum,
    TableFormatEnum,
    TargetStoreEnum,
    ThroughputModeEnum,
)

# Feature Definition helpers (local)
from sagemaker.mlops.feature_store.feature_definition import (
    FeatureDefinition,
    FeatureTypeEnum,
    CollectionTypeEnum,
    FractionalFeatureDefinition,
    IntegralFeatureDefinition,
    StringFeatureDefinition,
    ListCollectionType,
    SetCollectionType,
    VectorCollectionType,
)

# Utility functions (local)
from sagemaker.mlops.feature_store.feature_utils import (
    as_hive_ddl,
    create_athena_query,
    get_session_from_role,
    ingest_dataframe,
    load_feature_definitions_from_dataframe,
)

# Classes (local)
from sagemaker.mlops.feature_store.athena_query import AthenaQuery
from sagemaker.mlops.feature_store.dataset_builder import (
    DatasetBuilder,
    FeatureGroupToBeMerged,
    JoinComparatorEnum,
    JoinTypeEnum,
    TableType,
)
from sagemaker.mlops.feature_store.ingestion_manager_pandas import (
    IngestionError,
    IngestionManagerPandas,
)

__all__ = [
    # Resources
    "FeatureGroup",
    "FeatureMetadata",
    # Shapes
    "DataCatalogConfig",
    "FeatureParameter",
    "FeatureValue",
    "Filter",
    "OfflineStoreConfig",
    "OnlineStoreConfig",
    "OnlineStoreSecurityConfig",
    "S3StorageConfig",
    "SearchExpression",
    "ThroughputConfig",
    "TtlDuration",
    # Enums
    "DeletionModeEnum",
    "ExpirationTimeResponseEnum",
    "FilterOperatorEnum",
    "OnlineStoreStorageTypeEnum",
    "ResourceEnum",
    "SearchOperatorEnum",
    "SortOrderEnum",
    "TableFormatEnum",
    "TargetStoreEnum",
    "ThroughputModeEnum",
    # Feature Definitions
    "FeatureDefinition",
    "FeatureTypeEnum",
    "CollectionTypeEnum",
    "FractionalFeatureDefinition",
    "IntegralFeatureDefinition",
    "StringFeatureDefinition",
    "ListCollectionType",
    "SetCollectionType",
    "VectorCollectionType",
    # Utility functions
    "as_hive_ddl",
    "create_athena_query",
    "get_session_from_role",
    "ingest_dataframe",
    "load_feature_definitions_from_dataframe",
    # Classes
    "AthenaQuery",
    "DatasetBuilder",
    "FeatureGroupToBeMerged",
    "IngestionError",
    "IngestionManagerPandas",
    "JoinComparatorEnum",
    "JoinTypeEnum",
    "TableType",
]