Merge branch 'aws:master' into master

Author: SifeiLi

CHANGELOG.md

@@ -1,5 +1,17 @@
 # Changelog
 
+## v2.207.1 (2024-02-06)
+
+### Bug Fixes and Other Changes
+
+ * Add PT 2.1 as a supported framework for the smdistributed distribution
+ * Enable private docker registry support for ModelBuilder
+ * HF PT 2.1 Image Configs
+
+### Documentation Changes
+
+ * add setup commands for documentation generation
+
 ## v2.207.0 (2024-02-05)
 
 ### Features

VERSION

@@ -1 +1 @@
-2.207.1.dev0
+2.207.2.dev0

doc/api/prep_data/feature_store.rst

@@ -91,6 +91,10 @@ Inputs
     :members:
     :show-inheritance:
 
+.. autoclass:: sagemaker.feature_store.inputs.TtlDuration
+    :members:
+    :show-inheritance:
+
 .. autoclass:: sagemaker.feature_store.inputs.S3StorageConfig
     :members:
     :show-inheritance:

src/sagemaker/estimator.py

@@ -3285,7 +3285,6 @@ class Framework(EstimatorBase):
     UNSUPPORTED_DLC_IMAGE_FOR_SM_PARALLELISM = (
         "2.0.1-gpu-py310-cu121",
         "2.0-gpu-py310-cu121",
-        "2.1.0-gpu-py310",
     )
 
     def __init__(
@@ -3959,7 +3958,7 @@ def _distribution_configuration(self, distribution):
             for unsupported_image in Framework.UNSUPPORTED_DLC_IMAGE_FOR_SM_PARALLELISM:
                 if (
                     unsupported_image in img_uri and not torch_distributed_enabled
-                ):  # disabling DLC images with CUDA12
+                ):  # disabling DLC images without SMDataParallel or SMModelParallel
                     raise ValueError(
                         f"SMDistributed is currently incompatible with DLC image: {img_uri}. "
                         "(Could be due to CUDA version being greater than 11.)"