add 2.1 unsupported smddp

sirutBuasai · knikure · commit ed13e4d43374 · 2024-01-10T09:50:28.000-08:00
diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py
@@ -3282,7 +3282,7 @@ class Framework(EstimatorBase):
     """
 
     _framework_name = None
-    UNSUPPORTED_DLC_IMAGE_FOR_SM_PARALLELISM = ("2.0.1-gpu-py310-cu121", "2.0-gpu-py310-cu121")
+    UNSUPPORTED_DLC_IMAGE_FOR_SM_PARALLELISM = ("2.0.1-gpu-py310-cu121", "2.0-gpu-py310-cu121", "2.1.0-gpu-py310")
 
     def __init__(
         self,
