formatting

ruhanprasad · ruhanprasad · commit e450813951bc · 2024-03-08T19:40:34.000Z
diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py
@@ -3282,9 +3282,7 @@ class Framework(EstimatorBase):
     """
 
     _framework_name = None
-    UNSUPPORTED_DLC_IMAGE_FOR_SM_PARALLELISM = (
-        "2.0.1-gpu-py310-cu121",
-    )
+    UNSUPPORTED_DLC_IMAGE_FOR_SM_PARALLELISM = ("2.0.1-gpu-py310-cu121",)
 
     def __init__(
         self,
diff --git a/src/sagemaker/fw_utils.py b/src/sagemaker/fw_utils.py
@@ -161,8 +161,6 @@
     "2.2.0",
 ]
 
-
-
 TORCH_DISTRIBUTED_GPU_SUPPORTED_FRAMEWORK_VERSIONS = [
     "1.13.1",
     "2.0.0",

Original file line number	Diff line number	Diff line change
`@@ -161,8 +161,6 @@`
`161`	`161`	`"2.2.0",`
`162`	`162`	`]`
`163`	`163`
`164`		`-`
`165`		`-`
`166`	`164`	`TORCH_DISTRIBUTED_GPU_SUPPORTED_FRAMEWORK_VERSIONS = [`
`167`	`165`	`"1.13.1",`
`168`	`166`	`"2.0.0",`