Skip to content

Commit bd4414a

Browse files
pranavvm26pranavvm26
authored
Merge branch 'aws:master' into master
2 parents 4b419aa + 31f34dd commit bd4414a

Some content is hidden

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

50 files changed

+1754
-372
lines changed

‎CHANGELOG.md

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

3+
## v2.247.1 (2025-06-23)
4+
5+
### Bug Fixes and Other Changes
6+
7+
* update image_uri_configs 06-19-2025 07:18:34 PST
8+
9+
## v2.247.0 (2025-06-13)
10+
11+
### Features
12+
13+
* Add support for MetricDefinitions in ModelTrainer
14+
15+
### Bug Fixes and Other Changes
16+
17+
* update jumpstart region_config, update image_uri_configs 06-12-2025 07:18:12 PST
18+
* Add ignore_patterns in ModelTrainer to ignore specific files/folders
19+
* Allow import failure for internal _hashlib module
20+
21+
## v2.246.0 (2025-06-04)
22+
23+
### Features
24+
25+
* Triton v25.04 DLC
26+
27+
### Bug Fixes and Other Changes
28+
29+
* Update Attrs version to widen support
30+
* update estimator documentation regarding hyperparameters for source_dir
31+
32+
## v2.245.0 (2025-05-28)
33+
34+
### Features
35+
36+
* Correct mypy type checking through PEP 561
37+
38+
### Bug Fixes and Other Changes
39+
40+
* MLFLow update for dependabot
41+
* addWaiterTimeoutHandling
42+
* merge method inputs with class inputs
43+
* update image_uri_configs 05-20-2025 07:18:17 PST
44+
45+
## v2.244.2 (2025-05-19)
46+
47+
### Bug Fixes and Other Changes
48+
49+
* include model channel for gated uncompressed models
50+
* clarify model monitor one time schedule bug
51+
* update jumpstart region_config 05-15-2025 07:18:15 PST
52+
* update image_uri_configs 05-14-2025 07:18:16 PST
53+
* Add image configs and region config for TPE (ap-east-2)
54+
* Improve defaults handling in ModelTrainer
55+
56+
## v2.244.1 (2025-05-15)
57+
58+
### Bug Fixes and Other Changes
59+
60+
* Fix Flask-Limiter version
61+
* Fix test_huggingface_tei_uris()
62+
* huggingface-llm-neuronx dlc
63+
* huggingface-neuronx dlc image_uri
64+
* huggingface-tei dlc image_uri
65+
* Fix test_deploy_with_update_endpoint()
66+
* add AG v1.3
67+
* parameter mismatch in update_endpoint
68+
* remove --strip-component for untar source tar.gz
69+
* Fix type annotations
70+
* chore: Allow omegaconf >=2.2,<3
71+
* honor json serialization of HPs
72+
* Map llama models to correct script
73+
* pin test dependency
74+
* fix bad initialization script error message
75+
* Improve error logging and documentation for issue 4007
76+
* build(deps): bump scikit-learn
77+
* build(deps): bump mlflow
78+
* build(deps): bump mlflow in /tests/data/serve_resources/mlflow/pytorch
79+
* chore: Add tei 1.6.0 image
80+
381
## v2.244.0 (2025-05-02)
482

583
### Features

‎VERSION

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

‎pyproject.toml

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ classifiers = [
3131
"Programming Language :: Python :: 3.12",
3232
]
3333
dependencies = [
34-
"attrs>=23.1.0,<24",
34+
"attrs>=24,<26",
3535
"boto3>=1.35.75,<2.0",
3636
"cloudpickle>=2.2.1",
3737
"docker",
@@ -40,12 +40,12 @@ dependencies = [
4040
"importlib-metadata>=1.4.0,<7.0",
4141
"jsonschema",
4242
"numpy==1.26.4",
43-
"omegaconf>=2.2,<=2.3",
43+
"omegaconf>=2.2,<3",
4444
"packaging>=23.0,<25",
4545
"pandas",
4646
"pathos",
4747
"platformdirs",
48-
"protobuf>=3.12,<6.0",
48+
"protobuf>=3.12,<6.32",
4949
"psutil",
5050
"PyYAML>=6.0.1",
5151
"requests",
@@ -55,7 +55,8 @@ dependencies = [
5555
"tblib>=1.7.0,<4",
5656
"tqdm",
5757
"urllib3>=1.26.8,<3.0.0",
58-
"uvicorn"
58+
"uvicorn",
59+
"graphene>=3,<4"
5960
]
6061

6162
[project.scripts]

‎requirements/extras/test_requirements.txt

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,8 @@ stopit==1.1.2
1616
# Update tox.ini to have correct version of airflow constraints file
1717
apache-airflow==2.10.4
1818
apache-airflow-providers-amazon==7.2.1
19-
attrs>=23.1.0,<24
19+
Flask-Limiter==3.11
20+
attrs>=24,<26
2021
fabric==3.2.2
2122
requests==2.32.2
2223
sagemaker-experiments==0.1.35
@@ -43,7 +44,7 @@ nbformat>=5.9,<6
4344
accelerate>=0.24.1,<=0.27.0
4445
schema==0.7.5
4546
tensorflow>=2.16.2,<=2.18.0
46-
mlflow>=2.12.2,<2.13
47+
mlflow>=2.14.2,<3
4748
huggingface_hub==0.26.2
4849
uvicorn>=0.30.1
4950
fastapi==0.115.4

‎setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def get_optional_dependencies():
5858
version=HERE.joinpath("VERSION").read_text().strip(),
5959
packages=find_packages("src"),
6060
package_dir={"": "src"},
61-
package_data={"": ["*.whl"]},
61+
package_data={"": ["*.whl", "py.typed"]},
6262
py_modules=[os.path.splitext(os.path.basename(path))[0] for path in glob("src/*.py")],
6363
include_package_data=True,
6464
install_requires=get_dependencies(),

‎src/sagemaker/estimator.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -456,6 +456,9 @@ def __init__(
456456
A dictionary containing the hyperparameters to
457457
initialize this estimator with. (Default: None).
458458
459+
If a source directory is specified, the set_hyperparameters method escapes
460+
the dict argument as JSON, and updates the private hyperparameter attribute.
461+
459462
.. caution::
460463
You must not include any security-sensitive information, such as
461464
account access IDs, secrets, and tokens, in the dictionary for configuring

‎src/sagemaker/feature_store/feature_processor/_input_offset_parser.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,14 +72,16 @@ def get_offset_datetime(self, offset: Optional[str]) -> datetime:
7272

7373
return self.now + offset_td
7474

75-
def get_offset_date_year_month_day_hour(self, offset: Optional[str]) -> Tuple[str]:
75+
def get_offset_date_year_month_day_hour(
76+
self, offset: Optional[str]
77+
) -> Tuple[str, str, str, str]:
7678
"""Get the year, month, day and hour based on offset diff.
7779
7880
Args:
7981
offset (Optional[str]): Offset that is used for target date calcluation.
8082
8183
Returns:
82-
Tuple[str]: A tuple that consists of extracted year, month, day, hour from offset date.
84+
Tuple[str, str, str, str]: A tuple that consists of extracted year, month, day, hour from offset date.
8385
"""
8486
if offset is None:
8587
return (None, None, None, None)

‎src/sagemaker/image_uri_config/autogluon.json

Lines changed: 88 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,8 @@
1313
"0.8": "0.8.2",
1414
"1.0": "1.0.0",
1515
"1.1": "1.1.1",
16-
"1.2": "1.2.0"
16+
"1.2": "1.2.0",
17+
"1.3": "1.3.0"
1718
},
1819
"versions": {
1920
"0.3.1": {
@@ -605,6 +606,47 @@
605606
"py_versions": [
606607
"py311"
607608
]
609+
},
610+
"1.3.0": {
611+
"registries": {
612+
"af-south-1": "626614931356",
613+
"il-central-1": "780543022126",
614+
"ap-east-1": "871362719292",
615+
"ap-northeast-1": "763104351884",
616+
"ap-northeast-2": "763104351884",
617+
"ap-northeast-3": "364406365360",
618+
"ap-south-1": "763104351884",
619+
"ap-southeast-1": "763104351884",
620+
"ap-southeast-2": "763104351884",
621+
"ap-southeast-3": "907027046896",
622+
"ap-southeast-4": "457447274322",
623+
"ca-central-1": "763104351884",
624+
"eu-central-1": "763104351884",
625+
"eu-north-1": "763104351884",
626+
"eu-west-1": "763104351884",
627+
"eu-west-2": "763104351884",
628+
"eu-west-3": "763104351884",
629+
"eu-south-1": "692866216735",
630+
"me-south-1": "217643126080",
631+
"sa-east-1": "763104351884",
632+
"us-east-1": "763104351884",
633+
"us-east-2": "763104351884",
634+
"us-gov-east-1": "446045086412",
635+
"us-gov-west-1": "442386744353",
636+
"us-iso-east-1": "886529160074",
637+
"us-isob-east-1": "094389454867",
638+
"us-west-1": "763104351884",
639+
"us-west-2": "763104351884",
640+
"ca-west-1": "204538143572"
641+
},
642+
"repository": "autogluon-training",
643+
"processors": [
644+
"cpu",
645+
"gpu"
646+
],
647+
"py_versions": [
648+
"py311"
649+
]
608650
}
609651
}
610652
},
@@ -618,7 +660,8 @@
618660
"0.8": "0.8.2",
619661
"1.0": "1.0.0",
620662
"1.1": "1.1.1",
621-
"1.2": "1.2.0"
663+
"1.2": "1.2.0",
664+
"1.3": "1.3.0"
622665
},
623666
"versions": {
624667
"0.3.1": {
@@ -1243,6 +1286,49 @@
12431286
"py_versions": [
12441287
"py311"
12451288
]
1289+
},
1290+
"1.3.0": {
1291+
"registries": {
1292+
"af-south-1": "626614931356",
1293+
"il-central-1": "780543022126",
1294+
"ap-east-1": "871362719292",
1295+
"ap-northeast-1": "763104351884",
1296+
"ap-northeast-2": "763104351884",
1297+
"ap-northeast-3": "364406365360",
1298+
"ap-south-1": "763104351884",
1299+
"ap-southeast-1": "763104351884",
1300+
"ap-southeast-2": "763104351884",
1301+
"ap-southeast-3": "907027046896",
1302+
"ap-southeast-4": "457447274322",
1303+
"ca-central-1": "763104351884",
1304+
"cn-north-1": "727897471807",
1305+
"cn-northwest-1": "727897471807",
1306+
"eu-central-1": "763104351884",
1307+
"eu-north-1": "763104351884",
1308+
"eu-west-1": "763104351884",
1309+
"eu-west-2": "763104351884",
1310+
"eu-west-3": "763104351884",
1311+
"eu-south-1": "692866216735",
1312+
"me-south-1": "217643126080",
1313+
"sa-east-1": "763104351884",
1314+
"us-east-1": "763104351884",
1315+
"us-east-2": "763104351884",
1316+
"us-gov-east-1": "446045086412",
1317+
"us-gov-west-1": "442386744353",
1318+
"us-iso-east-1": "886529160074",
1319+
"us-isob-east-1": "094389454867",
1320+
"us-west-1": "763104351884",
1321+
"us-west-2": "763104351884",
1322+
"ca-west-1": "204538143572"
1323+
},
1324+
"repository": "autogluon-inference",
1325+
"processors": [
1326+
"cpu",
1327+
"gpu"
1328+
],
1329+
"py_versions": [
1330+
"py311"
1331+
]
12461332
}
12471333
}
12481334
}

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

Lines changed: 54 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
"inf2"
55
],
66
"version_aliases": {
7-
"0.0": "0.0.27"
7+
"0.0": "0.0.28"
88
},
99
"versions": {
1010
"0.0.16": {
@@ -589,6 +589,59 @@
589589
"container_version": {
590590
"inf2": "ubuntu22.04"
591591
}
592+
},
593+
"0.0.28": {
594+
"py_versions": [
595+
"py310"
596+
],
597+
"registries": {
598+
"af-south-1": "626614931356",
599+
"ap-east-1": "871362719292",
600+
"ap-east-2": "975050140332",
601+
"ap-northeast-1": "763104351884",
602+
"ap-northeast-2": "763104351884",
603+
"ap-northeast-3": "364406365360",
604+
"ap-south-1": "763104351884",
605+
"ap-south-2": "772153158452",
606+
"ap-southeast-1": "763104351884",
607+
"ap-southeast-2": "763104351884",
608+
"ap-southeast-3": "907027046896",
609+
"ap-southeast-4": "457447274322",
610+
"ap-southeast-5": "550225433462",
611+
"ap-southeast-7": "590183813437",
612+
"ca-central-1": "763104351884",
613+
"ca-west-1": "204538143572",
614+
"cn-north-1": "727897471807",
615+
"cn-northwest-1": "727897471807",
616+
"eu-central-1": "763104351884",
617+
"eu-central-2": "380420809688",
618+
"eu-north-1": "763104351884",
619+
"eu-south-1": "692866216735",
620+
"eu-south-2": "503227376785",
621+
"eu-west-1": "763104351884",
622+
"eu-west-2": "763104351884",
623+
"eu-west-3": "763104351884",
624+
"il-central-1": "780543022126",
625+
"me-central-1": "914824155844",
626+
"me-south-1": "217643126080",
627+
"mx-central-1": "637423239942",
628+
"sa-east-1": "763104351884",
629+
"us-east-1": "763104351884",
630+
"us-east-2": "763104351884",
631+
"us-gov-east-1": "446045086412",
632+
"us-gov-west-1": "442386744353",
633+
"us-iso-east-1": "886529160074",
634+
"us-isob-east-1": "094389454867",
635+
"us-isof-east-1": "303241398832",
636+
"us-isof-south-1": "454834333376",
637+
"us-west-1": "763104351884",
638+
"us-west-2": "763104351884"
639+
},
640+
"tag_prefix": "2.1.2-optimum0.0.28",
641+
"repository": "huggingface-pytorch-tgi-inference",
642+
"container_version": {
643+
"inf2": "ubuntu22.04"
644+
}
592645
}
593646
}
594647
}

‎src/sagemaker/image_uri_config/huggingface-neuron.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
],
1818
"repository": "huggingface-pytorch-inference-neuron",
1919
"registries": {
20+
"ap-east-2": "975050140332",
2021
"ap-northeast-1": "763104351884",
2122
"ap-south-1": "763104351884",
2223
"ap-south-2": "772153158452",

0 commit comments

Comments
 (0)