Feat/zimmer v2 backport - #6153
Open
Rishabh0255 wants to merge 1 commit into
Open
Conversation
Backports the 8 Zimmer pipeline step classes from v3 (PR aws#6140) to v2: - EndpointConfigStep, EndpointStep (SageMaker inference deployment) - InferenceComponentStep (multi-model endpoint support) - BedrockCustomModelStep, BedrockCustomModelDeploymentStep, BedrockModelImportStep, BedrockProvisionedModelThroughputStep - LineageStep (ML governance tracking) Also adds 8 new values to StepTypeEnum. Design: each step accepts an opaque 'arguments: Dict[str, Any]' that is forwarded to the pipeline service verbatim, mirroring the existing SDK convention used by LambdaStep and CallbackStep. Refer to the corresponding AWS Create* API references for accepted fields. Retryability per Zimmer step contract: only EndpointConfigStep is retryable. Cacheability: EndpointConfigStep and EndpointStep are structurally cacheable via cache_config. sim: https://issues.amazon.com/issues/P424919850 --- X-AI-Prompt: Backport the Zimmer pipeline step types from the v3 PR (aws#6140) to the v2 SDK on master-v2 X-AI-Tool: kiro-cli
Rishabh0255
requested a deployment
to
manual-approval
August 4, 2026 19:26 — with 
GitHub Actions
Waiting
GitHub Actions
Waiting
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
Backports the 8 Zimmer pipeline step classes from v3 (#6140) to v2.
EndpointConfigStep,EndpointStep,InferenceComponentStepBedrockCustomModelStep,BedrockCustomModelDeploymentStep,BedrockModelImportStep,BedrockProvisionedModelThroughputStepLineageStepAlso adds 8 values to
StepTypeEnumand mirrors the v3 test coverage.Relationship to v3 PR
Same design, same API surface, same test structure as #6140 -- only difference is import paths (v2's single-package
sagemaker.workflow.*layout vs v3'ssagemaker.mlops.workflow.*).Each step accepts an opaque
arguments: Dict[str, Any]that is forwarded to the pipeline service verbatim, matching the existing SDK convention used byLambdaStepandCallbackStep.EndpointConfigStepsupportsretry_policies.EndpointConfigStepandEndpointStepsupportcache_config.Testing
Unit tests (
tests/unit/sagemaker/workflow/test_zimmer_steps.py, 24 tests): construction,to_request()shape, enum values, retry-policy rejection on non-retryable steps, Bedrock properties, LineageStep argument validation.Integration test (
tests/integ/sagemaker/workflow/test_zimmer_lineage_step.py): full end-to-end LineageStep pipeline execution, assertsSucceededand verifies the created action ARN in step metadata.Additional gamma validation (performed against v3, applies unchanged to v2 -- same wire format, same server-side deserializer contract):
CreatePipeline+DescribePipelineround-trip for all 8 step typesSucceededon gammaCompatibility
No changes to existing v2 step types,
Pipeline, or the steps compiler beyond adding 8 values toStepTypeEnum.