Skip to content

Commit 9d5dc2c

Browse files
authored
Merge branch 'master' into framework-processor-custom-entrypoint
2 parents 1ea0cbb + 59ea1ab commit 9d5dc2c

181 files changed

Lines changed: 40693 additions & 47987 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,3 +42,4 @@ sagemaker_train/src/**/container_drivers/sourcecode.json
4242
sagemaker_train/src/**/container_drivers/distributed.json
4343
.kiro
4444
docs/api/generated/
45+
.hypothesis

.readthedocs.yaml

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,7 @@ build:
1111
sphinx:
1212
configuration: docs/conf.py
1313

14-
formats:
15-
- pdf
16-
- epub
14+
formats: []
1715

1816
python:
1917
install:

CHANGELOG.md

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,48 @@
11
# Changelog
2+
## v3.10.0 (2026-05-01)
3+
4+
### New Features
5+
- Make _PipelineExecution a public class
6+
- Add CodeArtifact support for ModelTrainer and FrameworkProcessor requirements.txt installation
7+
8+
### Bug Fixes
9+
- Fix S3 bucket operations
10+
- Fix potential S3 path traversal
11+
- Wire FrameworkProcessor code_location into code upload paths
12+
- Improve subprocess exception handling in git_utils
13+
14+
### Other
15+
- Update service-2.json with latest public botocore service model
16+
17+
## v3.9.0 (2026-04-23)
18+
19+
### New Features
20+
- **Train**: Add `wait_timeout` parameter to `train()` for SFT, DPO, RLAIF, RLVR, and BaseTrainer
21+
- **Evaluate**: Add MLflow experiment link to eval output
22+
- **JumpStart**: Allow `SAGEMAKER_HUB_NAME` environment variable to override the `HUB_NAME` constant
23+
24+
### Bug Fixes
25+
- **HyperparameterTuner**: Pass through full `OutputDataConfig` from `ModelTrainer` so `kms_key_id`, `compression_type`, and other fields are preserved
26+
- **HyperparameterTuner / ModelTrainer**: Propagate environment variables that were previously dropped
27+
- **sagemaker-core**: Improve error messages for waiter timeouts
28+
- **ModelBuilder**: Stop overwriting user-provided `HF_MODEL_ID` for DJL Serving
29+
- **ModelBuilder**: Keep `/opt/ml/model` writable when using `source_code` with DJL LMI
30+
- **Evaluate**: Skip `None` hyperparameters in `to_dict` instead of converting them to the string `"None"`
31+
- **Nova**: Add `us-west-2` to Nova supported regions
32+
- **DJL LMI**: Update ISO account mappings
33+
34+
## v3.8.0 (2026-04-16)
35+
36+
### New Features
37+
- **Feature Group Manager**: Feature Group Manager support
38+
- **Image Upgrades**: Image upgrades
39+
40+
### Bug Fixes
41+
- **ModelBuilder**: Add MLFlowConfig to Base Model
42+
- **Docker**: Support for docker compose > v2
43+
- **HuggingFace**: Improve SDK v3 Hugging Face support
44+
- **Dependencies**: Remove Pytorch hard dependency
45+
246
## v3.7.1 (2026-03-31)
347

448
### Features

VERSION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
3.7.1
1+
3.10.0

docs/api/sagemaker_mlops.rst

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -21,19 +21,18 @@ Workflow Management
2121
:undoc-members:
2222
:show-inheritance:
2323

24-
Local Development
25-
-----------------
24+
Feature Store
25+
-------------
2626

27-
.. automodule:: sagemaker.mlops.local
27+
.. automodule:: sagemaker.mlops.feature_store
2828
:members:
2929
:undoc-members:
3030
:show-inheritance:
3131

32+
Local Development
33+
-----------------
3234

33-
Feature Store
34-
-------------
35-
36-
.. automodule:: sagemaker.mlops.feature_store
35+
.. automodule:: sagemaker.mlops.local
3736
:members:
3837
:undoc-members:
3938
:show-inheritance:

docs/conf.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,25 @@
105105
'sagemaker.serve.model_server.torchserve.inference',
106106
'sagemaker.serve.model_server.torchserve.xgboost_inference',
107107
'sagemaker.serve.model_server.triton.model',
108+
# Heavy runtime dependencies — mock to avoid slow/failing imports during doc build
109+
'boto3',
110+
'botocore',
111+
'torch',
112+
'tensorflow',
113+
'sklearn',
114+
'numpy',
115+
'pandas',
116+
'scipy',
117+
'xgboost',
118+
'lightgbm',
119+
'mlflow',
120+
'pyspark',
121+
'transformers',
122+
'datasets',
123+
'trl',
108124
]
109125

110126
suppress_warnings = ['autodoc.import_error']
127+
128+
# Speed up builds: only regenerate autosummary stubs if they don't exist
129+
autosummary_generate_overwrite = False

pyproject.toml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,10 +33,10 @@ classifiers = [
3333
"Programming Language :: Python :: 3.12",
3434
]
3535
dependencies = [
36-
"sagemaker-core>=2.7.1,<3.0.0",
37-
"sagemaker-train>=1.7.1,<2.0.0",
38-
"sagemaker-serve>=1.7.1,<2.0.0",
39-
"sagemaker-mlops>=1.7.1,<2.0.0",
36+
"sagemaker-core>=2.10.0,<3.0.0",
37+
"sagemaker-train>=1.10.0,<2.0.0",
38+
"sagemaker-serve>=1.10.0,<2.0.0",
39+
"sagemaker-mlops>=1.10.0,<2.0.0",
4040
]
4141

4242
[project.optional-dependencies]

requirements/extras/test_requirements.txt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,4 +12,6 @@ omegaconf
1212
graphene
1313
typing_extensions>=4.9.0
1414
tensorflow>=2.16.2,<=2.19.0
15-
build
15+
build
16+
docker>=5.0.2,<8.0
17+
filelock>=3.0.0

sagemaker-core/CHANGELOG.md

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,40 @@
11
# Changelog
2+
## v2.10.0 (2026-05-01)
3+
4+
### New Features
5+
6+
- Add CodeArtifact support for ModelTrainer and FrameworkProcessor requirements.txt installation
7+
8+
### Bug Fixes
9+
10+
- Fix S3 bucket operations
11+
- Fix potential S3 path traversal
12+
- Wire FrameworkProcessor code_location into code upload paths
13+
- Improve subprocess exception handling in git_utils
14+
15+
### Other
16+
17+
- Update service-2.json with latest public botocore service model
18+
19+
## v2.9.0 (2026-04-23)
20+
21+
### Bug Fixes
22+
23+
- **Waiters**: Improve error messages for waiter timeouts
24+
- **DJL LMI**: Update ISO account mappings in image URI config
25+
26+
## v2.8.0 (2026-04-16)
27+
28+
### New Features
29+
30+
- **Image Upgrades**: Updated image URI configurations
31+
32+
### Bug Fixes
33+
34+
- **ModelBuilder**: Add MLFlowConfig to Base Model
35+
- **Local Mode**: Support for docker compose > v2
36+
- **Dependencies**: Remove Pytorch hard dependency
37+
238
## v2.7.1 (2026-03-31)
339

440
### Features

sagemaker-core/VERSION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
2.7.1
1+
2.10.0

0 commit comments

Comments
 (0)