Merge branch 'master' into fix/jumpstart-amt-tracking

Author: evakravi

doc/requirements.txt

@@ -1,4 +1,4 @@
-sphinx==3.1.1
+sphinx==3.1.2
 sphinx-rtd-theme==0.5.0
 docutils==0.15.2
 packaging==20.9

src/sagemaker/huggingface/model.py

@@ -304,6 +304,7 @@ def register(
         approval_status=None,
         description=None,
         drift_check_baselines=None,
+        customer_metadata_properties=None,
         domain=None,
     ):
         """Creates a model package for creating SageMaker models or listing on Marketplace.
@@ -332,6 +333,8 @@ def register(
                 or "PendingManualApproval". Defaults to ``PendingManualApproval``.
             description (str): Model Package description. Defaults to ``None``.
             drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
+            customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
+                metadata properties (default: None).
             domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
                 "MACHINE_LEARNING" (default: None).
 
@@ -362,6 +365,7 @@ def register(
             approval_status,
             description,
             drift_check_baselines=drift_check_baselines,
+            customer_metadata_properties=customer_metadata_properties,
             domain=domain,
         )
 

src/sagemaker/sklearn/model.py

@@ -151,6 +151,8 @@ def register(
         marketplace_cert=False,
         approval_status=None,
         description=None,
+        drift_check_baselines=None,
+        customer_metadata_properties=None,
         domain=None,
     ):
         """Creates a model package for creating SageMaker models or listing on Marketplace.
@@ -176,6 +178,9 @@ def register(
             approval_status (str): Model Approval Status, values can be "Approved", "Rejected",
                 or "PendingManualApproval" (default: "PendingManualApproval").
             description (str): Model Package description (default: None).
+            drift_check_baselines (DriftCheckBaselines): DriftCheckBaselines object (default: None).
+            customer_metadata_properties (dict[str, str]): A dictionary of key-value paired
+                metadata properties (default: None).
             domain (str): Domain values can be "COMPUTER_VISION", "NATURAL_LANGUAGE_PROCESSING",
                 "MACHINE_LEARNING" (default: None).
 
@@ -205,6 +210,8 @@ def register(
             marketplace_cert,
             approval_status,
             description,
+            drift_check_baselines=drift_check_baselines,
+            customer_metadata_properties=customer_metadata_properties,
             domain=domain,
         )
 

src/sagemaker/workflow/pipeline_context.py

@@ -16,6 +16,7 @@
 import warnings
 import inspect
 from typing import Dict
+from functools import wraps
 
 from sagemaker.session import Session, SessionSettings
 
@@ -132,6 +133,7 @@ def runnable_by_pipeline(run_func):
     The job will be started during pipeline execution.
     """
 
+    @wraps(run_func)
     def wrapper(*args, **kwargs):
         self_instance = args[0]
         if isinstance(self_instance.sagemaker_session, PipelineSession):