Merge branch 'master' into master

Author: mwfongAWS

src/sagemaker/config/config_schema.py

@@ -106,6 +106,7 @@
 LOCAL_CODE = "local_code"
 SERVING_PORT = "serving_port"
 CONTAINER_CONFIG = "container_config"
+CONTAINER_ROOT = "container_root"
 REGION_NAME = "region_name"
 
 
@@ -1090,6 +1091,9 @@ def _simple_path(*args: str):
                 SERVING_PORT: {
                     TYPE: "integer",
                 },
+                CONTAINER_ROOT: {
+                    TYPE: "string",
+                },
                 CONTAINER_CONFIG: {
                     TYPE: OBJECT,
                 },

src/sagemaker/jumpstart/payload_utils.py

@@ -37,12 +37,12 @@ class PayloadSerializer:
 
     def __init__(
         self,
-        bucket: str = get_jumpstart_content_bucket(),
+        bucket: Optional[str] = None,
         region: str = JUMPSTART_DEFAULT_REGION_NAME,
         s3_client: Optional[boto3.client] = None,
     ) -> None:
         """Initializes PayloadSerializer object."""
-        self.bucket = bucket
+        self.bucket = bucket or get_jumpstart_content_bucket()
         self.region = region
         self.s3_client = s3_client