Merge branch 'master' into processing-job-codeartifact-support

Author: akuma12

CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## v2.192.1 (2023-10-13)
+
+### Bug Fixes and Other Changes
+
+ * update local mode schema
+ * import error in unsupported js regions
+ * Update Ec2 instance type to g5.4xlarge in test_huggingface_torch_distributed.py
+
 ## v2.192.0 (2023-10-11)
 
 ### Features

VERSION

@@ -1 +1 @@
-2.192.1.dev0
+2.192.2.dev0

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
 

tests/integ/test_huggingface_torch_distributed.py

@@ -32,7 +32,7 @@ def test_huggingface_torch_distributed_g5_glue(
             transformers_version=huggingface_training_latest_version,
             pytorch_version=huggingface_training_pytorch_latest_version,
             instance_count=1,
-            instance_type="ml.g5.8xlarge",
+            instance_type="ml.g5.4xlarge",
             hyperparameters={
                 "model_name_or_path": "distilbert-base-cased",
                 "task_name": "wnli",