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)