feature: add SageMaker FeatureStore support (aws#476) (aws#501) (aws#528)

Author: icywang86rui

setup.py

@@ -47,6 +47,7 @@ def read_version():
 # Specific use case dependencies
 extras = {
     "analytics": ["pandas"],
+    "feature-store": ["pandas"],
     "local": [
         "urllib3>=1.21.1,<1.26,!=1.25.0,!=1.25.1",
         "docker-compose>=1.25.2",

src/sagemaker/feature_store/__init__.py

@@ -0,0 +1,12 @@
+# Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.

src/sagemaker/feature_store/feature_definition.py

@@ -0,0 +1,99 @@
+# Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+"""The Feature Definitions for FeatureStore."""
+from __future__ import absolute_import
+
+from enum import Enum
+from typing import Dict, Any
+
+import attr
+
+from sagemaker.feature_store.inputs import Config
+
+
+class FeatureTypeEnum(Enum):
+    """Enum of feature types."""
+
+    FRACTIONAL = "Fractional"
+    INTEGRAL = "Integral"
+    STRING = "String"
+
+
+@attr.s
+class FeatureDefinition(Config):
+    """Feature definition.
+
+    Attributes:
+        feature_name (str): The name of the feature
+        feature_type (FeatureTypeEnum): The type of the feature
+    """
+
+    feature_name: str = attr.ib()
+    feature_type: FeatureTypeEnum = attr.ib()
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Constructs a dictionary based on the attributes"""
+        return Config.construct_dict(
+            FeatureName=self.feature_name, FeatureType=self.feature_type.value
+        )
+
+
+class FractionalFeatureDefinition(FeatureDefinition):
+    """Fractional feature definition.
+
+    Attributes:
+        feature_name (str): The name of the feature
+        feature_type (FeatureTypeEnum): A `FeatureTypeEnum.FRACTIONAL` type
+    """
+
+    def __init__(self, feature_name: str):
+        """Constructs an instance of FractionalFeatureDefinition.
+
+        Args:
+            feature_name (str): the name of the feature.
+        """
+        super(FractionalFeatureDefinition, self).__init__(feature_name, FeatureTypeEnum.FRACTIONAL)
+
+
+class IntegralFeatureDefinition(FeatureDefinition):
+    """Fractional feature definition.
+
+    Attributes:
+        feature_name (str): the name of the feature.
+        feature_type (FeatureTypeEnum): a `FeatureTypeEnum.INTEGRAL` type.
+    """
+
+    def __init__(self, feature_name: str):
+        """Constructs an instance of IntegralFeatureDefinition.
+
+        Args:
+            feature_name (str): the name of the feature.
+        """
+        super(IntegralFeatureDefinition, self).__init__(feature_name, FeatureTypeEnum.INTEGRAL)
+
+
+class StringFeatureDefinition(FeatureDefinition):
+    """Fractional feature definition.
+
+    Attributes:
+        feature_name (str): the name of the feature.
+        feature_type (FeatureTypeEnum): a `FeatureTypeEnum.STRING` type.
+    """
+
+    def __init__(self, feature_name: str):
+        """Constructs an instance of StringFeatureDefinition.
+
+        Args:
+            feature_name (str): the name of the feature.
+        """
+        super(StringFeatureDefinition, self).__init__(feature_name, FeatureTypeEnum.STRING)