Skip to content

feature: Feat: Feature Store in Sagemaker SDK v3 (5496)#5716

Draft
aviruthen wants to merge 1 commit intoaws:masterfrom
aviruthen:feature/feat-feature-store-in-sagemaker-sdk-v3-5496
Draft

feature: Feat: Feature Store in Sagemaker SDK v3 (5496)#5716
aviruthen wants to merge 1 commit intoaws:masterfrom
aviruthen:feature/feat-feature-store-in-sagemaker-sdk-v3-5496

Conversation

@aviruthen
Copy link
Copy Markdown
Collaborator

Description

Feature Store already exists in SDK V3 under sagemaker-mlops at sagemaker.mlops.feature_store with a comprehensive migration guide. The issue is discoverability: (1) the sagemaker.mlops.init.py doesn't mention or re-export feature_store, (2) the init.py imports FeatureGroup and FeatureMetadata from core but does NOT import FeatureStore (needed for search()), (3) the create_dataset() convenience function referenced in the MIGRATION_GUIDE.md doesn't actually exist in feature_utils.py or anywhere else, (4) there is no top-level documentation or entry point directing users to the new location. The fix involves: exposing feature_store in the mlops init.py, adding the missing create_dataset function, adding FeatureStore to the feature_store init.py re-exports, and ensuring the migration guide references match actual code.

Related Issue

Related issue: 5496

Changes Made

  • sagemaker-mlops/src/sagemaker/mlops/__init__.py
  • sagemaker-mlops/src/sagemaker/mlops/feature_store/__init__.py
  • sagemaker-mlops/src/sagemaker/mlops/feature_store/feature_utils.py
  • sagemaker-mlops/src/sagemaker/mlops/feature_store/MIGRATION_GUIDE.md
  • sagemaker-mlops/tests/unit/sagemaker/mlops/feature_store/test_feature_store_init.py

AI-Generated PR

This PR was automatically generated by the PySDK Issue Agent.

  • Confidence score: 85%
  • Classification: type: feature request
  • SDK version target: V3

Merge Checklist

  • Changes are backward compatible
  • Commit message follows prefix: description format
  • Unit tests added/updated
  • Integration tests added (if applicable)
  • Documentation updated (if applicable)

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.

1 participant