Skip to content

Commit 55917a7

Browse files
Merge branch 'aws:master' into master
2 parents d28a265 + a7baead commit 55917a7

File tree

62 files changed

+3300
-214
lines changed

Some content is hidden

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

62 files changed

+3300
-214
lines changed

CHANGELOG.md

Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,55 @@
11
# Changelog
22

3+
## v2.239.0 (2025-02-01)
4+
5+
### Features
6+
7+
* Add support for deepseek recipes
8+
9+
### Bug Fixes and Other Changes
10+
11+
* mpirun protocol - distributed training with @remote decorator
12+
* Allow telemetry only in supported regions
13+
* Fix ssh host policy
14+
15+
## v2.238.0 (2025-01-29)
16+
17+
### Features
18+
19+
* use jumpstart deployment config image as default optimization image
20+
21+
### Bug Fixes and Other Changes
22+
23+
* chore: add new images for HF TGI
24+
* update image_uri_configs 01-29-2025 06:18:08 PST
25+
* skip TF tests for unsupported versions
26+
* Merge branch 'master-rba' into local_merge
27+
* Add missing attributes to local resourceconfig
28+
* update image_uri_configs 01-27-2025 06:18:13 PST
29+
* update image_uri_configs 01-24-2025 06:18:11 PST
30+
* add missing schema definition in docs
31+
* Omegaconf upgrade
32+
* SageMaker @remote function: Added multi-node functionality
33+
* remove option
34+
* fix typo
35+
* fix tests
36+
* Add an option for user to remove inputs and container artifacts when using local model trainer
37+
38+
## v2.237.3 (2025-01-09)
39+
40+
### Bug Fixes and Other Changes
41+
42+
* pin metadata-version to 2.3
43+
* model server might have already done a serialization. honor that by not decoding the request again if it is not already bytes or bytestream
44+
* Disable jumpstart tests missing clean up logic
45+
* Jumpstart ap southeast 5
46+
* add autogluon 1.2
47+
* updated inference script to cover context
48+
* security update -> use sha256 instead of md5 for file hashing
49+
* Fix Flake8 Violations
50+
* Added parsing string support for situations where custom code might be used (ie. mlflow)
51+
* Updating Inference Optimization Validations
52+
353
## v2.237.2 (2024-12-17)
454

555
### Bug Fixes and Other Changes

VERSION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
2.237.3.dev0
1+
2.239.1.dev0

doc/overview.rst

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2565,6 +2565,9 @@ set default values for. For the full schema, see `sagemaker.config.config_schema
25652565
      KmsKeyId: 'kmskeyid10'
25662566
    TransformResources:
25672567
      VolumeKmsKeyId: 'volumekmskeyid4'
2568+
Tags:
2569+
    - Key: 'tag_key'
2570+
      Value: 'tag_value
25682571
  CompilationJob:
25692572
  # https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateCompilationJob.html
25702573
    OutputConfig:

pyproject.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ dependencies = [
4040
"importlib-metadata>=1.4.0,<7.0",
4141
"jsonschema",
4242
"numpy>=1.9.0,<2.0",
43-
"omegaconf>=2.2,<2.3",
43+
"omegaconf>=2.2,<=2.3",
4444
"packaging>=20.0",
4545
"pandas",
4646
"pathos",
@@ -73,13 +73,15 @@ pattern = "(?P<version>.+)"
7373
[tool.hatch.metadata.hooks.custom]
7474

7575
[tool.hatch.build.targets.wheel]
76+
core-metadata-version = "2.3"
7677
packages = ["src/sagemaker"]
7778
exclude = ["src/sagemaker/serve/model_server/triton/pack_conda_env.sh"]
7879

7980
[tool.hatch.build.targets.wheel.shared-scripts]
8081
"src/sagemaker/serve/model_server/triton/pack_conda_env.sh" = "pack_conda_env.sh"
8182

8283
[tool.hatch.build.targets.sdist]
84+
core-metadata-version = "2.3"
8385
only-include = [
8486
"/requirements/extras",
8587
"/src",

src/sagemaker/fw_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,7 @@
152152
"2.1.0",
153153
"2.1.2",
154154
"2.2.0",
155+
"2.3.0",
155156
"2.3.1",
156157
"2.4.1",
157158
]

src/sagemaker/image_uri_config/autogluon.json

Lines changed: 88 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,8 @@
1212
"0.7": "0.7.0",
1313
"0.8": "0.8.2",
1414
"1.0": "1.0.0",
15-
"1.1": "1.1.1"
15+
"1.1": "1.1.1",
16+
"1.2": "1.2.0"
1617
},
1718
"versions": {
1819
"0.3.1": {
@@ -563,6 +564,47 @@
563564
"py_versions": [
564565
"py311"
565566
]
567+
},
568+
"1.2.0": {
569+
"registries": {
570+
"af-south-1": "626614931356",
571+
"il-central-1": "780543022126",
572+
"ap-east-1": "871362719292",
573+
"ap-northeast-1": "763104351884",
574+
"ap-northeast-2": "763104351884",
575+
"ap-northeast-3": "364406365360",
576+
"ap-south-1": "763104351884",
577+
"ap-southeast-1": "763104351884",
578+
"ap-southeast-2": "763104351884",
579+
"ap-southeast-3": "907027046896",
580+
"ap-southeast-4": "457447274322",
581+
"ca-central-1": "763104351884",
582+
"eu-central-1": "763104351884",
583+
"eu-north-1": "763104351884",
584+
"eu-west-1": "763104351884",
585+
"eu-west-2": "763104351884",
586+
"eu-west-3": "763104351884",
587+
"eu-south-1": "692866216735",
588+
"me-south-1": "217643126080",
589+
"sa-east-1": "763104351884",
590+
"us-east-1": "763104351884",
591+
"us-east-2": "763104351884",
592+
"us-gov-east-1": "446045086412",
593+
"us-gov-west-1": "442386744353",
594+
"us-iso-east-1": "886529160074",
595+
"us-isob-east-1": "094389454867",
596+
"us-west-1": "763104351884",
597+
"us-west-2": "763104351884",
598+
"ca-west-1": "204538143572"
599+
},
600+
"repository": "autogluon-training",
601+
"processors": [
602+
"cpu",
603+
"gpu"
604+
],
605+
"py_versions": [
606+
"py311"
607+
]
566608
}
567609
}
568610
},
@@ -575,7 +617,8 @@
575617
"0.7": "0.7.0",
576618
"0.8": "0.8.2",
577619
"1.0": "1.0.0",
578-
"1.1": "1.1.1"
620+
"1.1": "1.1.1",
621+
"1.2": "1.2.0"
579622
},
580623
"versions": {
581624
"0.3.1": {
@@ -1157,6 +1200,49 @@
11571200
"py_versions": [
11581201
"py311"
11591202
]
1203+
},
1204+
"1.2.0": {
1205+
"registries": {
1206+
"af-south-1": "626614931356",
1207+
"il-central-1": "780543022126",
1208+
"ap-east-1": "871362719292",
1209+
"ap-northeast-1": "763104351884",
1210+
"ap-northeast-2": "763104351884",
1211+
"ap-northeast-3": "364406365360",
1212+
"ap-south-1": "763104351884",
1213+
"ap-southeast-1": "763104351884",
1214+
"ap-southeast-2": "763104351884",
1215+
"ap-southeast-3": "907027046896",
1216+
"ap-southeast-4": "457447274322",
1217+
"ca-central-1": "763104351884",
1218+
"cn-north-1": "727897471807",
1219+
"cn-northwest-1": "727897471807",
1220+
"eu-central-1": "763104351884",
1221+
"eu-north-1": "763104351884",
1222+
"eu-west-1": "763104351884",
1223+
"eu-west-2": "763104351884",
1224+
"eu-west-3": "763104351884",
1225+
"eu-south-1": "692866216735",
1226+
"me-south-1": "217643126080",
1227+
"sa-east-1": "763104351884",
1228+
"us-east-1": "763104351884",
1229+
"us-east-2": "763104351884",
1230+
"us-gov-east-1": "446045086412",
1231+
"us-gov-west-1": "442386744353",
1232+
"us-iso-east-1": "886529160074",
1233+
"us-isob-east-1": "094389454867",
1234+
"us-west-1": "763104351884",
1235+
"us-west-2": "763104351884",
1236+
"ca-west-1": "204538143572"
1237+
},
1238+
"repository": "autogluon-inference",
1239+
"processors": [
1240+
"cpu",
1241+
"gpu"
1242+
],
1243+
"py_versions": [
1244+
"py311"
1245+
]
11601246
}
11611247
}
11621248
}

src/sagemaker/image_uri_config/huggingface-llm-neuronx.json

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
"ap-southeast-1": "763104351884",
1919
"ap-southeast-2": "763104351884",
2020
"ap-southeast-4": "457447274322",
21+
"ap-southeast-5": "550225433462",
2122
"cn-north-1": "727897471807",
2223
"cn-northwest-1": "727897471807",
2324
"eu-central-1": "763104351884",
@@ -51,6 +52,7 @@
5152
"ap-southeast-1": "763104351884",
5253
"ap-southeast-2": "763104351884",
5354
"ap-southeast-4": "457447274322",
55+
"ap-southeast-5": "550225433462",
5456
"cn-north-1": "727897471807",
5557
"cn-northwest-1": "727897471807",
5658
"eu-central-1": "763104351884",
@@ -84,6 +86,7 @@
8486
"ap-southeast-1": "763104351884",
8587
"ap-southeast-2": "763104351884",
8688
"ap-southeast-4": "457447274322",
89+
"ap-southeast-5": "550225433462",
8790
"cn-north-1": "727897471807",
8891
"cn-northwest-1": "727897471807",
8992
"eu-central-1": "763104351884",
@@ -117,6 +120,7 @@
117120
"ap-southeast-1": "763104351884",
118121
"ap-southeast-2": "763104351884",
119122
"ap-southeast-4": "457447274322",
123+
"ap-southeast-5": "550225433462",
120124
"cn-north-1": "727897471807",
121125
"cn-northwest-1": "727897471807",
122126
"eu-central-1": "763104351884",
@@ -150,6 +154,7 @@
150154
"ap-southeast-1": "763104351884",
151155
"ap-southeast-2": "763104351884",
152156
"ap-southeast-4": "457447274322",
157+
"ap-southeast-5": "550225433462",
153158
"cn-north-1": "727897471807",
154159
"cn-northwest-1": "727897471807",
155160
"eu-central-1": "763104351884",
@@ -183,6 +188,7 @@
183188
"ap-southeast-1": "763104351884",
184189
"ap-southeast-2": "763104351884",
185190
"ap-southeast-4": "457447274322",
191+
"ap-southeast-5": "550225433462",
186192
"cn-north-1": "727897471807",
187193
"cn-northwest-1": "727897471807",
188194
"eu-central-1": "763104351884",
@@ -216,6 +222,7 @@
216222
"ap-southeast-1": "763104351884",
217223
"ap-southeast-2": "763104351884",
218224
"ap-southeast-4": "457447274322",
225+
"ap-southeast-5": "550225433462",
219226
"cn-north-1": "727897471807",
220227
"cn-northwest-1": "727897471807",
221228
"eu-central-1": "763104351884",
@@ -247,6 +254,7 @@
247254
"ap-southeast-1": "763104351884",
248255
"ap-southeast-2": "763104351884",
249256
"ap-southeast-4": "457447274322",
257+
"ap-southeast-5": "550225433462",
250258
"cn-north-1": "727897471807",
251259
"cn-northwest-1": "727897471807",
252260
"eu-central-1": "763104351884",
@@ -280,6 +288,7 @@
280288
"ap-southeast-1": "763104351884",
281289
"ap-southeast-2": "763104351884",
282290
"ap-southeast-4": "457447274322",
291+
"ap-southeast-5": "550225433462",
283292
"cn-north-1": "727897471807",
284293
"cn-northwest-1": "727897471807",
285294
"eu-central-1": "763104351884",
@@ -313,6 +322,7 @@
313322
"ap-southeast-1": "763104351884",
314323
"ap-southeast-2": "763104351884",
315324
"ap-southeast-4": "457447274322",
325+
"ap-southeast-5": "550225433462",
316326
"cn-north-1": "727897471807",
317327
"cn-northwest-1": "727897471807",
318328
"eu-central-1": "763104351884",

0 commit comments

Comments
 (0)