sagemaker-python-sdk icon indicating copy to clipboard operation
sagemaker-python-sdk copied to clipboard

Support `numpy >= 1.26.4` and Understanding Pinned Dependencies in General

Open siddk-lbm opened this issue 2 months ago • 1 comments

Describe the feature you'd like

Right now, the sagemaker-python-sdk pins certain dependencies to extremely old and restrictive versions. The biggest offender is numpy == 1.26.4 which is over 18 months old, making sagemaker incompatible with many existing codebases.

How would this feature be used? Please describe.

Minimally, would be nice to unpin numpy (permitting latest versions), and in general, understanding why current sagemaker dependencies are pinned to the versions they are currently (e.g., protobuf).

Describe alternatives you've considered

We're considering building our own version of the Sagemaker wheel (with unpinned dependencies), but not sure which of these dependencies will actually break/result in new behavior.

Any help with this would be much appreciated! Thanks!

siddk-lbm avatar Oct 07 '25 14:10 siddk-lbm

Looks to be in the works as per: https://github.com/aws/sagemaker-python-sdk/issues/4882

des1-gner avatar Oct 09 '25 21:10 des1-gner