Skip to content

Feature store V3#5554

Draft
BassemHalim wants to merge 8 commits intoaws:feature-store-v3from
BassemHalim:feature-store-v3
Draft

Feature store V3#5554
BassemHalim wants to merge 8 commits intoaws:feature-store-v3from
BassemHalim:feature-store-v3

Conversation

@BassemHalim
Copy link

Issue #, if available:

Migrate Sagemaker Feature Store to V3

Description of changes:

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

adishaa and others added 8 commits January 16, 2026 07:00
- Add LakeFormationConfig class to configure Lake Formation governance on offline stores
- Implement FeatureGroup subclass with Lake Formation integration capabilities
- Add helper methods for S3 URI/ARN conversion and Lake Formation role management
- Add S3 deny policy generation for Lake Formation access control
- Implement Lake Formation resource registration and S3 bucket policy setup
- Add integration tests for Lake Formation feature store workflows
- Add unit tests for Lake Formation configuration and policy generation
- Update feature_store module exports to include FeatureGroup and LakeFormationConfig
- Update API documentation to include Feature Store section in sagemaker_mlops.rst
- Enable fine-grained access control for feature store offline stores using AWS Lake Formation
- Most code was ported from V2 and the imports were updated
- Some usage of session.describe_feature_group was converted to FeatureGroup.get() from sagemaker.core which calls describe_feature_Group under the hood
- The feature_scheduler to_pipeline was refactored to replace Estimator from V2 to use ModelTrainer from V3. (Estimator was removed from V3)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants