Feature store v3 (#5539)

* Feature store v3 (#5490)

* feat: Add Feature Store Support to V3

* Add feature store tests

---------

Co-authored-by: adishaa <adishaa@amazon.com>

* Changes:
- Add feature_store_functions_report.md documenting all 63 functions
  across feature_store module (excluding feature_processor)
- Add comprehensive unit tests for get_feature_group_as_dataframe:
  * Session handling (provided, from region, from role)
  * Error cases (missing session/region, missing event_time)
  * Latest ingestion logic with event time
  * Query string manipulation and table placeholder
  * Verbose and silent logging modes
  * Kwargs passing to as_dataframe
- Add comprehensive unit tests for prepare_fg_from_dataframe_or_file:
  * DataFrame and file path input handling
  * Session/region/role configuration
  * Record ID creation and validation
  * Event ID creation with timestamp
  * Duplicate record detection
  * Column name formatting
  * CSV kwargs passing
  * Feature definition loading

* Fix

* Fix

* fix

* Integ tests

* Fix

* Add feature store telemetry

* Bug fix for GetRecordResponse not printable and error message for ingest_dataframe

* Fix: DatasetBuilder.to_dataframe() docstring and instantiation

* Fix unit test failures

* fix unit test

---------

Co-authored-by: Aditi Sharma <165942273+Aditi2424@users.noreply.github.com>
Co-authored-by: adishaa <adishaa@amazon.com>
Co-authored-by: Molly He <mollyhe@amazon.com>

Author: 4 people

sagemaker-core/src/sagemaker/core/telemetry/constants.py

@@ -28,6 +28,7 @@ class Feature(Enum):
     MODEL_TRAINER = 14
     MODEL_CUSTOMIZATION = 15
     MLOPS = 16
+    FEATURE_STORE = 17
 
     def __str__(self):  # pylint: disable=E0307
         """Return the feature name."""

sagemaker-core/src/sagemaker/core/telemetry/telemetry_logging.py

@@ -56,6 +56,7 @@
     str(Feature.MODEL_TRAINER): 14,
     str(Feature.MODEL_CUSTOMIZATION): 15,
     str(Feature.MLOPS): 16,
+    str(Feature.FEATURE_STORE): 17,
 }
 
 STATUS_TO_CODE = {
@@ -81,6 +82,9 @@ def wrapper(*args, **kwargs):
             if len(args) > 0 and hasattr(args[0], "sagemaker_session"):
                 # Get the sagemaker_session from the instance method args
                 sagemaker_session = args[0].sagemaker_session
+            elif len(args) > 0 and hasattr(args[0], "_sagemaker_session"):
+                # Get the sagemaker_session from the instance method args (private attribute)
+                sagemaker_session = args[0]._sagemaker_session
             elif feature == Feature.REMOTE_FUNCTION:
                 # Get the sagemaker_session from the function keyword arguments for remote function
                 sagemaker_session = kwargs.get(

sagemaker-core/src/sagemaker/core/utils/utils.py

@@ -298,6 +298,22 @@ def __new__(cls):
             cls._instance = super().__new__(cls)
         return cls._instance
 
+    def __repr__(self):
+        """Return clean representation for debugging."""
+        return "Unassigned()"
+
+    def __str__(self):
+        """Return empty string for clean printing."""
+        return ""
+
+    def __iter__(self):
+        """Return empty iterator to prevent iteration errors."""
+        return iter([])
+
+    def __bool__(self):
+        """Return False for truthiness checks."""
+        return False
+
 
 class SingletonMeta(type):
     """

sagemaker-core/tests/unit/generated/test_utils.py

@@ -371,3 +371,50 @@ def test_serialize_method_nested_shape():
             "StringValue": "string",
         },
     }
+
+
+class TestUnassignedBehavior:
+    """Test Unassigned class methods for proper behavior.
+    
+    Bug fix: GetRecordResponse is not printable and cannot be parsed via iterator.
+    Error: TypeError: 'Unassigned' object is not iterable
+    """
+
+    def test_unassigned_repr(self):
+        """Test that Unassigned has clean repr."""
+        u = Unassigned()
+        assert repr(u) == "Unassigned()"
+
+    def test_unassigned_str(self):
+        """Test that Unassigned converts to empty string."""
+        u = Unassigned()
+        assert str(u) == ""
+
+    def test_unassigned_bool(self):
+        """Test that Unassigned is falsy."""
+        u = Unassigned()
+        assert not u
+        assert bool(u) is False
+
+    def test_unassigned_iter(self):
+        """Test that Unassigned is iterable and returns empty list."""
+        u = Unassigned()
+        result = list(u)
+        assert result == []
+
+    def test_unassigned_singleton(self):
+        """Test that Unassigned is a singleton."""
+        u1 = Unassigned()
+        u2 = Unassigned()
+        assert u1 is u2
+
+    def test_unassigned_in_conditional(self):
+        """Test that Unassigned works correctly in conditionals."""
+        u = Unassigned()
+        
+        # Should evaluate to False
+        if u:
+            pytest.fail("Unassigned should be falsy")
+        
+        # Should work with not
+        assert not u