Merge branch 'master' into feat/jumpstart-model-package-arn-instance-type-variants

Author: evakravi

CHANGELOG.md

@@ -1,5 +1,26 @@
 # Changelog
 
+## v2.192.0 (2023-10-11)
+
+### Features
+
+ * jumpstart estimator enable infra check flag
+ * jumpstart default payloads
+ * allow non-python files in job dependencies
+ * allow configuring docker container in local mode
+
+### Bug Fixes and Other Changes
+
+ * js tagging s3 prefix
+ * Batch transform: Add support for split_type == "None" in local mode
+ * use correct line endings and s3 uris on windows
+ * Fixed bug in _create_training_details
+ * DJL Neuronx 0.24.0
+
+### Documentation Changes
+
+ * Include FeatureGroup's load_feature_definitions API documentation
+
 ## v2.191.0 (2023-10-05)
 
 ### Features

VERSION

@@ -1 +1 @@
-2.191.1.dev0
+2.192.1.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",