Skip to content

Commit dba42d8

Browse files
beniericbenieric
authored
Merge branch 'master' into modernize-metadata
2 parents ecd7349 + b96c98e commit dba42d8

File tree

76 files changed

+2842
-277
lines changed

Some content is hidden

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

76 files changed

+2842
-277
lines changed

‎CHANGELOG.md

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

3+
## v2.224.0 (2024-06-19)
4+
5+
### Features
6+
7+
* JumpStartModel attach
8+
9+
### Bug Fixes and Other Changes
10+
11+
* feat(sagemaker-mlflow): New features for SageMaker MLflow
12+
* Upgrading to PT 2.3 for release
13+
* chore: use ml.g5.2xlarge for integ test
14+
* Enable telemetry logging for Remote function
15+
* Fix Dependabot Issues - MLFlow Version
16+
17+
## v2.223.0 (2024-06-13)
18+
19+
### Features
20+
21+
* add 'ModelCard' property to Register step
22+
23+
### Bug Fixes and Other Changes
24+
25+
* Fix Sniping bug fix
26+
* Implement custom telemetry logging in SDK
27+
* Fix ci unit-tests
28+
* update image_uri_configs 06-12-2024 07:17:03 PST
29+
30+
## v2.222.1 (2024-06-12)
31+
32+
### Bug Fixes and Other Changes
33+
34+
* First changes
35+
* estimator.deploy not respecting instance type
36+
37+
## v2.222.0 (2024-06-07)
38+
39+
### Features
40+
41+
* jumpstart telemetry
42+
43+
### Bug Fixes and Other Changes
44+
45+
* update image_uri_configs 06-06-2024 07:17:31 PST
46+
* bump requests from 2.31.0 to 2.32.2 in /requirements/extras
47+
* chore: add HF LLM neuronx 0.0.23 image
48+
* Updates for DJL 0.28.0 release
49+
* chore(deps): bump mlflow from 2.11.1 to 2.12.1 in /tests/data/serve_resources/mlflow/tensorflow
50+
* chore(deps): bump mlflow from 2.11.1 to 2.12.1 in /tests/data/serve_resources/mlflow/xgboost
51+
* chore(deps): bump mlflow from 2.10.2 to 2.12.1 in /tests/data/serve_resources/mlflow/pytorch
52+
* chore(deps): bump apache-airflow from 2.9.0 to 2.9.1 in /requirements/extras
53+
* chore(deps): bump requests from 2.31.0 to 2.32.2 in /tests/data/serve_resources/mlflow/pytorch
54+
* Fix ci unit-tests
55+
* Making project name in workflow files dynamic
56+
* update image_uri_configs 05-29-2024 07:17:35 PST
57+
* Update: SM Endpoint Routing Strategy Support.
58+
359
## v2.221.1 (2024-05-22)
460

561
### Bug Fixes and Other Changes

‎README.rst

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,13 @@ SageMaker Python SDK is tested on:
9595
- Python 3.10
9696
- Python 3.11
9797

98+
Telemetry
99+
~~~~~~~~~~~~~~~
100+
101+
The ``sagemaker`` library has telemetry enabled to help us better understand user needs, diagnose issues, and deliver new features. This telemetry tracks the usage of various SageMaker functions.
102+
103+
If you prefer to opt out of telemetry, you can easily do so by setting the ``TelemetryOptOut`` parameter to ``true`` in the SDK defaults configuration. For detailed instructions, please visit `Configuring and using defaults with the SageMaker Python SDK <https://sagemaker.readthedocs.io/en/stable/overview.html#configuring-and-using-defaults-with-the-sagemaker-python-sdk>`__.
104+
98105
AWS Permissions
99106
~~~~~~~~~~~~~~~
100107

‎VERSION

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

‎requirements/extras/test_requirements.txt

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,11 @@ awslogs==0.14.0
1212
black==24.3.0
1313
stopit==1.1.2
1414
# Update tox.ini to have correct version of airflow constraints file
15-
apache-airflow==2.9.0
15+
apache-airflow==2.9.2
1616
apache-airflow-providers-amazon==7.2.1
1717
attrs>=23.1.0,<24
1818
fabric==2.6.0
19-
requests==2.31.0
19+
requests==2.32.2
2020
sagemaker-experiments==0.1.35
2121
Jinja2==3.1.4
2222
pyvis==0.2.1
@@ -37,3 +37,4 @@ nbformat>=5.9,<6
3737
accelerate>=0.24.1,<=0.27.0
3838
schema==0.7.5
3939
tensorflow>=2.1,<=2.16
40+
mlflow>=2.12.2,<2.13

‎src/sagemaker/chainer/model.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,10 @@
2828
from sagemaker.model import FrameworkModel, MODEL_SERVER_WORKERS_PARAM_NAME
2929
from sagemaker.chainer import defaults
3030
from sagemaker.deserializers import NumpyDeserializer
31+
from sagemaker.model_card import (
32+
ModelCard,
33+
ModelPackageModelCard,
34+
)
3135
from sagemaker.predictor import Predictor
3236
from sagemaker.serializers import NumpySerializer
3337
from sagemaker.utils import to_string
@@ -175,6 +179,7 @@ def register(
175179
data_input_configuration: Optional[Union[str, PipelineVariable]] = None,
176180
skip_model_validation: Optional[Union[str, PipelineVariable]] = None,
177181
source_uri: Optional[Union[str, PipelineVariable]] = None,
182+
model_card: Optional[Union[ModelPackageModelCard, ModelCard]] = None,
178183
):
179184
"""Creates a model package for creating SageMaker models or listing on Marketplace.
180185
@@ -226,6 +231,8 @@ def register(
226231
validation. Values can be "All" or "None" (default: None).
227232
source_uri (str or PipelineVariable): The URI of the source for the model package
228233
(default: None).
234+
model_card (ModeCard or ModelPackageModelCard): document contains qualitative and
235+
quantitative information about a model (default: None).
229236
230237
Returns:
231238
str: A string of SageMaker Model Package ARN.
@@ -266,6 +273,7 @@ def register(
266273
data_input_configuration=data_input_configuration,
267274
skip_model_validation=skip_model_validation,
268275
source_uri=source_uri,
276+
model_card=model_card,
269277
)
270278

271279
def prepare_container_def(

‎src/sagemaker/estimator.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1724,6 +1724,7 @@ def register(
17241724
data_input_configuration=None,
17251725
skip_model_validation=None,
17261726
source_uri=None,
1727+
model_card=None,
17271728
**kwargs,
17281729
):
17291730
"""Creates a model package for creating SageMaker models or listing on Marketplace.
@@ -1772,6 +1773,8 @@ def register(
17721773
skip_model_validation (str): Indicates if you want to skip model validation.
17731774
Values can be "All" or "None" (default: None).
17741775
source_uri (str): The URI of the source for the model package (default: None).
1776+
model_card (ModeCard or ModelPackageModelCard): document contains qualitative and
1777+
quantitative information about a model (default: None).
17751778
**kwargs: Passed to invocation of ``create_model()``. Implementations may customize
17761779
``create_model()`` to accept ``**kwargs`` to customize model creation during
17771780
deploy. For more, see the implementation docs.
@@ -1817,6 +1820,7 @@ def register(
18171820
data_input_configuration=data_input_configuration,
18181821
skip_model_validation=skip_model_validation,
18191822
source_uri=source_uri,
1823+
model_card=model_card,
18201824
)
18211825

18221826
@property

‎src/sagemaker/huggingface/model.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,10 @@
2626
)
2727
from sagemaker.metadata_properties import MetadataProperties
2828
from sagemaker.model import FrameworkModel, MODEL_SERVER_WORKERS_PARAM_NAME
29+
from sagemaker.model_card import (
30+
ModelCard,
31+
ModelPackageModelCard,
32+
)
2933
from sagemaker.predictor import Predictor
3034
from sagemaker.serializers import JSONSerializer
3135
from sagemaker.session import Session
@@ -362,6 +366,7 @@ def register(
362366
data_input_configuration: Optional[Union[str, PipelineVariable]] = None,
363367
skip_model_validation: Optional[Union[str, PipelineVariable]] = None,
364368
source_uri: Optional[Union[str, PipelineVariable]] = None,
369+
model_card: Optional[Union[ModelPackageModelCard, ModelCard]] = None,
365370
):
366371
"""Creates a model package for creating SageMaker models or listing on Marketplace.
367372
@@ -414,6 +419,8 @@ def register(
414419
validation. Values can be "All" or "None" (default: None).
415420
source_uri (str or PipelineVariable): The URI of the source for the model package
416421
(default: None).
422+
model_card (ModeCard or ModelPackageModelCard): document contains qualitative and
423+
quantitative information about a model (default: None).
417424
418425
Returns:
419426
A `sagemaker.model.ModelPackage` instance.
@@ -462,6 +469,7 @@ def register(
462469
data_input_configuration=data_input_configuration,
463470
skip_model_validation=skip_model_validation,
464471
source_uri=source_uri,
472+
model_card=model_card,
465473
)
466474

467475
def prepare_container_def(

‎src/sagemaker/image_uri_config/djl-lmi.json

Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,39 @@
1+
{
2+
"scope": [
3+
"inference"
4+
],
5+
"versions": {
6+
"0.28.0": {
7+
"registries": {
8+
"af-south-1": "626614931356",
9+
"il-central-1": "780543022126",
10+
"ap-east-1": "871362719292",
11+
"ap-northeast-1": "763104351884",
12+
"ap-northeast-2": "763104351884",
13+
"ap-northeast-3": "364406365360",
14+
"ap-south-1": "763104351884",
15+
"ap-southeast-1": "763104351884",
16+
"ap-southeast-2": "763104351884",
17+
"ap-southeast-3": "907027046896",
18+
"ca-central-1": "763104351884",
19+
"cn-north-1": "727897471807",
20+
"cn-northwest-1": "727897471807",
21+
"eu-central-1": "763104351884",
22+
"eu-north-1": "763104351884",
23+
"eu-west-1": "763104351884",
24+
"eu-west-2": "763104351884",
25+
"eu-west-3": "763104351884",
26+
"eu-south-1": "692866216735",
27+
"me-south-1": "217643126080",
28+
"sa-east-1": "763104351884",
29+
"us-east-1": "763104351884",
30+
"us-east-2": "763104351884",
31+
"us-west-1": "763104351884",
32+
"us-west-2": "763104351884",
33+
"ca-west-1": "204538143572"
34+
},
35+
"repository": "djl-inference",
36+
"tag_prefix": "0.28.0-lmi10.0.0-cu124"
37+
}
38+
}
39+
}

‎src/sagemaker/image_uri_config/djl-neuronx.json

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,24 @@
33
"inference"
44
],
55
"versions": {
6+
"0.28.0": {
7+
"registries": {
8+
"ap-northeast-1": "763104351884",
9+
"ap-south-1": "763104351884",
10+
"ap-southeast-1": "763104351884",
11+
"ap-southeast-2": "763104351884",
12+
"eu-central-1": "763104351884",
13+
"eu-west-1": "763104351884",
14+
"eu-west-3": "763104351884",
15+
"sa-east-1": "763104351884",
16+
"us-east-1": "763104351884",
17+
"us-east-2": "763104351884",
18+
"us-west-2": "763104351884",
19+
"ca-west-1": "204538143572"
20+
},
21+
"repository": "djl-inference",
22+
"tag_prefix": "0.28.0-neuronx-sdk2.18.2"
23+
},
624
"0.27.0": {
725
"registries": {
826
"ap-northeast-1": "763104351884",

‎src/sagemaker/image_uri_config/djl-tensorrtllm.json

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,38 @@
33
"inference"
44
],
55
"versions": {
6+
"0.28.0": {
7+
"registries": {
8+
"af-south-1": "626614931356",
9+
"il-central-1": "780543022126",
10+
"ap-east-1": "871362719292",
11+
"ap-northeast-1": "763104351884",
12+
"ap-northeast-2": "763104351884",
13+
"ap-northeast-3": "364406365360",
14+
"ap-south-1": "763104351884",
15+
"ap-southeast-1": "763104351884",
16+
"ap-southeast-2": "763104351884",
17+
"ap-southeast-3": "907027046896",
18+
"ca-central-1": "763104351884",
19+
"cn-north-1": "727897471807",
20+
"cn-northwest-1": "727897471807",
21+
"eu-central-1": "763104351884",
22+
"eu-north-1": "763104351884",
23+
"eu-west-1": "763104351884",
24+
"eu-west-2": "763104351884",
25+
"eu-west-3": "763104351884",
26+
"eu-south-1": "692866216735",
27+
"me-south-1": "217643126080",
28+
"sa-east-1": "763104351884",
29+
"us-east-1": "763104351884",
30+
"us-east-2": "763104351884",
31+
"us-west-1": "763104351884",
32+
"us-west-2": "763104351884",
33+
"ca-west-1": "204538143572"
34+
},
35+
"repository": "djl-inference",
36+
"tag_prefix": "0.28.0-tensorrtllm0.9.0-cu122"
37+
},
638
"0.27.0": {
739
"registries": {
840
"af-south-1": "626614931356",

0 commit comments

Comments
 (0)