You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
Repository navigation
Commit cd53577
Browse filesBrowse the repository at this point in the historyBrowse files
fix(local): Align local mode with the SageMaker service (#6317)
Bring local mode behavior in line with the hosted SageMaker service:
- Runtime invoke_endpoint now returns ResponseMetadata and
InvokedProductionVariant, matching the boto3 runtime response.
- LocalSagemakerClient gains a describe_user_profile passthrough so
Studio role resolution works instead of raising AttributeError.
- MultiRecordStrategy buffers bytes or str based on the record type,
so local batch transform handles binary inputs.
- Processing jobs no longer require an IAM role in local mode, where
the role is never used.
- Local pipeline execution state is isolated per instance and per
session instead of living on class-level or client-level state.
- Local pipeline execution ids are 12-char uppercase alphanumerics,
matching the service format and downstream name limits. This ports
#5283 (merged to master-v2, by aviruthen) to the V3 codebase; see
#5269.
Response headers are looked up case-insensitively (containers send
Content-Type), and local/image.py imports DIR_PARAM_NAME and
SAGEMAKER_OUTPUT_LOCATION from sagemaker.core.constants instead of
referencing the never-imported sagemaker.serve.model_builder, which
made every local endpoint fail with AttributeError unless sagemaker.serve
happened to be imported first; three unit tests skipped for that reason
are re-enabled.
The role fix also skips expand_role() when building the create request,
since expand_role(None) raised TypeError once the constructor let a
missing role through. Local-mode integration tests (Docker) cover the
role-less processing job and the invoke_endpoint response contract.
Fixes#3348Fixes#4417Fixes#4996Fixes#5562Fixes#5572Fixes#5604
---
X-AI-Prompt: Fix S-effort PySDK V3 bugs, local-mode theme
X-AI-Tool: Kiro
0 commit comments