Sagemaker V2 to V3 migration.#5644

Closed

bhanutejagk wants to merge 38 commits intoaws:masterfrom

bhanutejagk:pytorch-2.10-currency

Commits on Feb 9, 2026

Add PyTorch 2.10 Training DLC with CUDA 13.0 and Python 3.13
Bhanu Teja Goshikonda
committed
Configure build for PyTorch 2.10 training EC2 images
Bhanu Teja Goshikonda
committed

Commits on Feb 10, 2026

fix: add setuptools for pkg_resources in Python 3.13 (OSS compliance)
Bhanu Teja Goshikonda
committed
Merge remote-tracking branch 'upstream/master' into pytorch-2.10-currency
Bhanu Teja Goshikonda
committed
fix: move setuptools install to EC2/SageMaker stages for pkg_resources
Bhanu Teja Goshikonda
committed
fix: pin setuptools to 81.0.0 for pkg_resources compatibility
Bhanu Teja Goshikonda
committed

Commits on Feb 11, 2026

fix: pin setuptools to 80.10.1 (pkg_resources removed in 81+)
Bhanu Teja Goshikonda
committed
fix: pin setuptools to 81.0.0 and remove redundant installs
Bhanu Teja Goshikonda
committed
Fix torch 2.10 version pinning and increase CPU image size baseline
Bhanu Teja Goshikonda
committed
Merge branch 'master' into pytorch-2.10-currency
bhanutejagk
authored
Fix torch 2.10 version pinning and remove setuptools pin
Bhanu Teja Goshikonda
committed

Commits on Feb 12, 2026

Revert pytorch install changes to match 2.9 style
Bhanu Teja Goshikonda
committed
Set build_inference to false
Bhanu Teja Goshikonda
committed
Remove fastai - requires torch<2.10, not compatible with PyTorch 2.10
Bhanu Teja Goshikonda
committed

Commits on Feb 18, 2026

Merge upstream/master into pytorch-2.10-currency
Bhanu Teja Goshikonda
committed
Migrate PyTorch 2.10 training tests from SageMaker SDK v2 to v3
Bhanu Teja Goshikonda
committed

Commits on Mar 10, 2026

Merge remote-tracking branch 'upstream/master' into pytorch-2.10-currency
Bhanu Teja Goshikonda
committed
Run tests.
Bhanu Teja Goshikonda
committed
Run tests.
Bhanu Teja Goshikonda
committed
Fix SageMaker v2 import errors for PyTorch 2.10 tests
Bhanu Teja Goshikonda
committed
Fix version fixture prefix collision in lookup_condition
Bhanu Teja Goshikonda
committed

Commits on Mar 11, 2026

Enable extended tests and apply black formatting
Bhanu Teja Goshikonda
committed
Merge remote-tracking branch 'upstream/master' into pytorch-2.10-currency
Bhanu Teja Goshikonda
committed
Disable sagemaker_benchmark_tests - all PT benchmarks are skipped and TF benchmarks use v2 SDK imports
Bhanu Teja Goshikonda
committed
Revert dlc_developer_config.toml to defaults
Bhanu Teja Goshikonda
committed
Wrap SM SDK v2 imports in try/except to prevent pytest collection failure under sagemaker>=3
Bhanu Teja Goshikonda
committed
Also wrap sagemaker.utils import in try/except for SM SDK v3 compatibility
Bhanu Teja Goshikonda
committed
fix: migrate sagemaker.exceptions imports for SM SDK v3 compatibility
Bhanu Teja Goshikonda
committed

Commits on Mar 12, 2026

Add SM SDK v3 test files for PyTorch 2.10, route v3 tests in sagemaker.py, configure EC2 buildspec
Bhanu Teja Goshikonda
committed
Add fastai back to 2.10 Dockerfiles (fastai 2.8.7 supports torch<3), rebuild image
Bhanu Teja Goshikonda
committed

Commits on Mar 13, 2026

Switch to SM buildspec with do_build=true, enable all SM tests (efa, rc, benchmark)
Bhanu Teja Goshikonda
committed
Fix sanity test failures for SM SDK v3: sagemaker version check, remote_function skip, pip_check mlflow/pandas exception, do_build=false
Bhanu Teja Goshikonda
committed
fix: sanity test compatibility for SM SDK v3
Bhanu Teja Goshikonda
committed
fix: SM v3 tests - fix sagemaker.modules imports and tighten boto bounds
Bhanu Teja Goshikonda
committed
Fix safety scan CVE allowlists, v3 requirements, and trigger rebuild
Bhanu Teja Goshikonda
committed
Fix v3 requirements: boto3>=1.42.2, mock>=4.0; run SM tests only
Bhanu Teja Goshikonda
committed
Fix v2/v3 SDK conflicts: conditional imports + standalone v3 timeout
Bhanu Teja Goshikonda
committed
fix: guard all v2-only sagemaker imports for SM SDK v3 compatibility
Bhanu Teja Goshikonda
committed