Skip to content

Commit 494e0d9

Browse files
bobbywlindseybobbywlindsey
authored
Merge branch 'master' into missing-register-params
2 parents 38b9114 + 8c93f92 commit 494e0d9

19 files changed

+1139
-2
lines changed

‎doc/amazon_sagemaker_model_building_pipeline.rst

+1,015
Large diffs are not rendered by default.

‎doc/index.rst

+11
Original file line numberDiff line numberDiff line change
@@ -102,3 +102,14 @@ You can use Amazon SageMaker Processing to perform data processing tasks such as
102102
:maxdepth: 2
103103

104104
amazon_sagemaker_processing
105+
106+
107+
*****************************************
108+
Amazon SageMaker Model Building Pipeline
109+
*****************************************
110+
You can use Amazon SageMaker Model Building Pipelines to orchestrate your machine learning workflow.
111+
112+
.. toctree::
113+
:maxdepth: 2
114+
115+
amazon_sagemaker_model_building_pipeline

‎doc/overview.rst

+10
Original file line numberDiff line numberDiff line change
@@ -1727,6 +1727,16 @@ For more information, see `SageMaker Workflow in Apache Airflow`_.
17271727
17281728
.. _SageMaker Workflow in Apache Airflow: https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/workflow/README.rst
17291729
1730+
************************************
1731+
SageMaker Model Building Pipeline
1732+
************************************
1733+
1734+
You can use Amazon SageMaker Model Building Pipelines to orchestrate your machine learning workflow.
1735+
1736+
For more information, see `SageMaker Model Building Pipeline`_.
1737+
1738+
.. _SageMaker Model Building Pipeline: https://github.com/aws/sagemaker-python-sdk/blob/master/doc/amazon_sagemaker_model_building_pipeline.rst
1739+
17301740
**************************
17311741
SageMaker Model Monitoring
17321742
**************************

‎doc/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
sphinx==3.1.1
1+
sphinx==3.1.2
22
sphinx-rtd-theme==0.5.0
33
docutils==0.15.2
44
packaging==20.9

‎doc/workflows/pipelines/sagemaker.workflow.pipelines.rst

+12
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ Execution Variables
5252
.. autoclass:: sagemaker.workflow.execution_variables.ExecutionVariable
5353

5454
.. autoclass:: sagemaker.workflow.execution_variables.ExecutionVariables
55+
:members: START_DATETIME, CURRENT_DATETIME, PIPELINE_EXECUTION_ID, PIPELINE_EXECUTION_ARN, PIPELINE_NAME, PIPELINE_ARN
5556

5657
Functions
5758
---------
@@ -73,6 +74,8 @@ Parameters
7374

7475
.. autoclass:: sagemaker.workflow.parameters.ParameterFloat
7576

77+
.. autoclass:: sagemaker.workflow.parameters.ParameterBoolean
78+
7679
Pipeline
7780
--------
7881

@@ -82,6 +85,13 @@ Pipeline
8285
.. autoclass:: sagemaker.workflow.pipeline._PipelineExecution
8386
:members:
8487

88+
Pipeline Context
89+
------------------
90+
91+
.. autoclass:: sagemaker.workflow.pipeline_context.PipelineSession
92+
:members:
93+
94+
8595
Parallelism Configuration
8696
-------------------------
8797

@@ -115,6 +125,8 @@ Step Collections
115125

116126
.. autoclass:: sagemaker.workflow.step_collections.EstimatorTransformer
117127

128+
.. autoclass:: sagemaker.workflow.model_step.ModelStep
129+
118130
Steps
119131
-----
120132

‎src/sagemaker/estimator.py

+4
Original file line numberDiff line numberDiff line change
@@ -1280,6 +1280,7 @@ def register(
12801280
model_name=None,
12811281
drift_check_baselines=None,
12821282
customer_metadata_properties=None,
1283+
domain=None,
12831284
**kwargs,
12841285
):
12851286
"""Creates a model package for creating SageMaker models or listing on Marketplace.
@@ -1311,6 +1312,8 @@ def register(
13111312
drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
13121313
customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
13131314
metadata properties (default: None).
1315+
domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
1316+
"MACHINE_LEARNING" (default: None).
13141317
**kwargs: Passed to invocation of ``create_model()``. Implementations may customize
13151318
``create_model()`` to accept ``**kwargs`` to customize model creation during
13161319
deploy. For more, see the implementation docs.
@@ -1342,6 +1345,7 @@ def register(
13421345
description,
13431346
drift_check_baselines=drift_check_baselines,
13441347
customer_metadata_properties=customer_metadata_properties,
1348+
domain=domain,
13451349
)
13461350

13471351
@property

‎src/sagemaker/huggingface/model.py

+4
Original file line numberDiff line numberDiff line change
@@ -305,6 +305,7 @@ def register(
305305
description=None,
306306
drift_check_baselines=None,
307307
customer_metadata_properties=None,
308+
domain=None,
308309
):
309310
"""Creates a model package for creating SageMaker models or listing on Marketplace.
310311
@@ -334,6 +335,8 @@ def register(
334335
drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
335336
customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
336337
metadata properties (default: None).
338+
domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
339+
"MACHINE_LEARNING" (default: None).
337340
338341
Returns:
339342
A `sagemaker.model.ModelPackage` instance.
@@ -363,6 +366,7 @@ def register(
363366
description,
364367
drift_check_baselines=drift_check_baselines,
365368
customer_metadata_properties=customer_metadata_properties,
369+
domain=domain,
366370
)
367371

368372
def prepare_container_def(

‎src/sagemaker/image_uri_config/sklearn.json

+31
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,37 @@
6363
"us-west-2": "246618743249"
6464
},
6565
"repository": "sagemaker-scikit-learn"
66+
},
67+
"1.0-1": {
68+
"py_versions": ["py3"],
69+
"registries": {
70+
"af-south-1": "510948584623",
71+
"ap-east-1": "651117190479",
72+
"ap-northeast-1": "354813040037",
73+
"ap-northeast-2": "366743142698",
74+
"ap-northeast-3": "867004704886",
75+
"ap-south-1": "720646828776",
76+
"ap-southeast-1": "121021644041",
77+
"ap-southeast-2": "783357654285",
78+
"ca-central-1": "341280168497",
79+
"cn-north-1": "450853457545",
80+
"cn-northwest-1": "451049120500",
81+
"eu-central-1": "492215442770",
82+
"eu-north-1": "662702820516",
83+
"eu-west-1": "141502667606",
84+
"eu-west-2": "764974769150",
85+
"eu-west-3": "659782779980",
86+
"eu-south-1": "978288397137",
87+
"me-south-1": "801668240914",
88+
"sa-east-1": "737474898029",
89+
"us-east-1": "683313688378",
90+
"us-east-2": "257758044811",
91+
"us-gov-west-1": "414596584902",
92+
"us-iso-east-1": "833128469047",
93+
"us-west-1": "746614075791",
94+
"us-west-2": "246618743249"
95+
},
96+
"repository": "sagemaker-scikit-learn"
6697
}
6798
}
6899
}

‎src/sagemaker/model.py

+4
Original file line numberDiff line numberDiff line change
@@ -309,6 +309,7 @@ def register(
309309
drift_check_baselines=None,
310310
customer_metadata_properties=None,
311311
validation_specification=None,
312+
domain=None,
312313
):
313314
"""Creates a model package for creating SageMaker models or listing on Marketplace.
314315
@@ -336,6 +337,8 @@ def register(
336337
drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
337338
customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
338339
metadata properties (default: None).
340+
domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
341+
"MACHINE_LEARNING" (default: None).
339342
340343
Returns:
341344
A `sagemaker.model.ModelPackage` instance.
@@ -365,6 +368,7 @@ def register(
365368
drift_check_baselines=drift_check_baselines,
366369
customer_metadata_properties=customer_metadata_properties,
367370
validation_specification=validation_specification,
371+
domain=domain,
368372
)
369373
model_package = self.sagemaker_session.create_model_package_from_containers(
370374
**model_pkg_args

‎src/sagemaker/mxnet/model.py

+4
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,7 @@ def register(
158158
description=None,
159159
drift_check_baselines=None,
160160
customer_metadata_properties=None,
161+
domain=None,
161162
):
162163
"""Creates a model package for creating SageMaker models or listing on Marketplace.
163164
@@ -185,6 +186,8 @@ def register(
185186
drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
186187
customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
187188
metadata properties (default: None).
189+
domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
190+
"MACHINE_LEARNING" (default: None).
188191
189192
Returns:
190193
A `sagemaker.model.ModelPackage` instance.
@@ -214,6 +217,7 @@ def register(
214217
description,
215218
drift_check_baselines=drift_check_baselines,
216219
customer_metadata_properties=customer_metadata_properties,
220+
domain=domain,
217221
)
218222

219223
def prepare_container_def(

‎src/sagemaker/pytorch/model.py

+4
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,7 @@ def register(
159159
description=None,
160160
drift_check_baselines=None,
161161
customer_metadata_properties=None,
162+
domain=None,
162163
):
163164
"""Creates a model package for creating SageMaker models or listing on Marketplace.
164165
@@ -186,6 +187,8 @@ def register(
186187
drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
187188
customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
188189
metadata properties (default: None).
190+
domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
191+
"MACHINE_LEARNING" (default: None).
189192
190193
Returns:
191194
A `sagemaker.model.ModelPackage` instance.
@@ -215,6 +218,7 @@ def register(
215218
description,
216219
drift_check_baselines=drift_check_baselines,
217220
customer_metadata_properties=customer_metadata_properties,
221+
domain=domain,
218222
)
219223

220224
def prepare_container_def(

‎src/sagemaker/session.py

+14
Original file line numberDiff line numberDiff line change
@@ -2803,6 +2803,7 @@ def create_model_package_from_containers(
28032803
drift_check_baselines=None,
28042804
customer_metadata_properties=None,
28052805
validation_specification=None,
2806+
domain=None,
28062807
):
28072808
"""Get request dictionary for CreateModelPackage API.
28082809
@@ -2830,6 +2831,8 @@ def create_model_package_from_containers(
28302831
drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
28312832
customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
28322833
metadata properties (default: None).
2834+
domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
2835+
"MACHINE_LEARNING" (default: None).
28332836
"""
28342837

28352838
model_pkg_request = get_create_model_package_request(
@@ -2848,6 +2851,7 @@ def create_model_package_from_containers(
28482851
drift_check_baselines=drift_check_baselines,
28492852
customer_metadata_properties=customer_metadata_properties,
28502853
validation_specification=validation_specification,
2854+
domain=domain,
28512855
)
28522856

28532857
def submit(request):
@@ -4218,6 +4222,7 @@ def get_model_package_args(
42184222
drift_check_baselines=None,
42194223
customer_metadata_properties=None,
42204224
validation_specification=None,
4225+
domain=None,
42214226
):
42224227
"""Get arguments for create_model_package method.
42234228
@@ -4248,6 +4253,8 @@ def get_model_package_args(
42484253
drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
42494254
customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
42504255
metadata properties (default: None).
4256+
domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
4257+
"MACHINE_LEARNING" (default: None).
42514258
Returns:
42524259
dict: A dictionary of method argument names and values.
42534260
"""
@@ -4289,6 +4296,8 @@ def get_model_package_args(
42894296
model_package_args["customer_metadata_properties"] = customer_metadata_properties
42904297
if validation_specification is not None:
42914298
model_package_args["validation_specification"] = validation_specification
4299+
if domain is not None:
4300+
model_package_args["domain"] = domain
42924301
return model_package_args
42934302

42944303

@@ -4309,6 +4318,7 @@ def get_create_model_package_request(
43094318
drift_check_baselines=None,
43104319
customer_metadata_properties=None,
43114320
validation_specification=None,
4321+
domain=None,
43124322
):
43134323
"""Get request dictionary for CreateModelPackage API.
43144324
@@ -4337,6 +4347,8 @@ def get_create_model_package_request(
43374347
drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
43384348
customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
43394349
metadata properties (default: None).
4350+
domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
4351+
"MACHINE_LEARNING" (default: None).
43404352
"""
43414353

43424354
if all([model_package_name, model_package_group_name]):
@@ -4362,6 +4374,8 @@ def get_create_model_package_request(
43624374
request_dict["CustomerMetadataProperties"] = customer_metadata_properties
43634375
if validation_specification:
43644376
request_dict["ValidationSpecification"] = validation_specification
4377+
if domain is not None:
4378+
request_dict["Domain"] = domain
43654379
if containers is not None:
43664380
if not all([content_types, response_types, inference_instances, transform_instances]):
43674381
raise ValueError(

‎src/sagemaker/sklearn/model.py

+4
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,7 @@ def register(
153153
description=None,
154154
drift_check_baselines=None,
155155
customer_metadata_properties=None,
156+
domain=None,
156157
):
157158
"""Creates a model package for creating SageMaker models or listing on Marketplace.
158159
@@ -180,6 +181,8 @@ def register(
180181
drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
181182
customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
182183
metadata properties (default: None).
184+
domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
185+
"MACHINE_LEARNING" (default: None).
183186
184187
Returns:
185188
A `sagemaker.model.ModelPackage` instance.
@@ -209,6 +212,7 @@ def register(
209212
description,
210213
drift_check_baselines=drift_check_baselines,
211214
customer_metadata_properties=customer_metadata_properties,
215+
domain=domain,
212216
)
213217

214218
def prepare_container_def(

‎src/sagemaker/tensorflow/model.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -205,6 +205,7 @@ def register(
205205
description=None,
206206
drift_check_baselines=None,
207207
customer_metadata_properties=None,
208+
domain=None,
208209
):
209210
"""Creates a model package for creating SageMaker models or listing on Marketplace.
210211
@@ -232,7 +233,8 @@ def register(
232233
drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
233234
customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
234235
metadata properties (default: None).
235-
236+
domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
237+
"MACHINE_LEARNING" (default: None).
236238
237239
Returns:
238240
A `sagemaker.model.ModelPackage` instance.
@@ -262,6 +264,7 @@ def register(
262264
description,
263265
drift_check_baselines=drift_check_baselines,
264266
customer_metadata_properties=customer_metadata_properties,
267+
domain=domain,
265268
)
266269

267270
def deploy(

0 commit comments

Comments
 (0)