Skip to content

Feature processor v3#5565

Draft
BassemHalim wants to merge 12 commits intoaws:masterfrom
BassemHalim:feature-processor-v3
Draft

Feature processor v3#5565
BassemHalim wants to merge 12 commits intoaws:masterfrom
BassemHalim:feature-processor-v3

Conversation

@BassemHalim
Copy link

@BassemHalim BassemHalim commented Feb 20, 2026

Port feature_processor module to SageMaker Python SDK v3

Migrates the feature_store.feature_processor module from sagemaker-python-sdk v2 to the v3 modular package structure (sagemaker-mlops), along with supporting changes in sagemaker-core.

Changes

Import path migration — All internal imports updated from sagemaker.feature_store.feature_processorsagemaker.mlops.feature_store.feature_processor, and external SDK dependencies remapped to their v3 locations:

  • sagemaker.Sessionsagemaker.core.helper.session_helper.Session
  • sagemaker.lineagesagemaker.core.lineage
  • sagemaker.remote_functionsagemaker.core.remote_function
  • sagemaker.workflowsagemaker.mlops.workflow
  • sagemaker.s3 / sagemaker.utils / sagemaker.vpc_utilssagemaker.core.*

Estimator → ModelTrainer migration (feature_scheduler.py) — Replaced the v2 Estimator dict-based construction with the v3 ModelTrainer API:

  • Uses Compute, Networking, StoppingCondition, SourceCode, OutputDataConfig, and Tag config objects
  • Creates a PipelineSession for pipeline-aware execution
  • TrainingStep now uses step_args from ModelTrainer.train() instead of estimator+inputs

Input channel format change (_config_uploader.py) — prepare_step_input now returns List[Channel] (using Channel/DataSource/S3DataSource shapes) instead of Dict[str, TrainingInput].

Session helper additions (sagemaker-core) — Added Feature Store methods to Session: delete_feature_group, describe_feature_group, create_feature_group, update_feature_group, and related config schema imports.

Diff with V2

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

@BassemHalim BassemHalim changed the base branch from master to feature-store-v3 February 25, 2026 21:40
@BassemHalim BassemHalim changed the base branch from feature-store-v3 to master February 27, 2026 05:19
@BassemHalim BassemHalim reopened this Feb 27, 2026
Aditi2424 and others added 7 commits February 27, 2026 11:19
* feat: Add Feature Store Support to V3

* Add feature store tests

---------

Co-authored-by: adishaa <adishaa@amazon.com>
Aditi2424 and others added 2 commits February 27, 2026 15:58
* feat: Add Feature Store Support to V3

* Add feature store tests

---------

Co-authored-by: adishaa <adishaa@amazon.com>
@BassemHalim BassemHalim deployed to manual-approval February 28, 2026 00:03 — with GitHub Actions Active
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.

4 participants