Fixed pylint

Author: karan6181

src/sagemaker/estimator.py

@@ -2607,9 +2607,8 @@ def _distribution_configuration(self, distribution):
             distribution_config[self.LAUNCH_SM_DDP_ENV_NAME] = smdataparallel_enabled
             distribution_config[self.INSTANCE_TYPE] = self.instance_type
             if smdataparallel_enabled:
-                distribution_config[self.SM_DDP_CUSTOM_MPI_OPTIONS] = smdistributed["dataparallel"].get(
-                    "custom_mpi_options", ""
-                )
+                distribution_config[self.SM_DDP_CUSTOM_MPI_OPTIONS] = \
+                    smdistributed["dataparallel"].get("custom_mpi_options", "")
 
         return distribution_config
 

tests/unit/test_estimator.py

@@ -120,8 +120,14 @@
 DISTRIBUTION_MPI_ENABLED = {
     "mpi": {"enabled": True, "custom_mpi_options": "options", "processes_per_host": 2}
 }
-DISTRIBUTION_SM_DDP_ENABLED = {"smdistributed": {"dataparallel": {"enabled": True,
-                                                                  "custom_mpi_options": "options"}}}
+DISTRIBUTION_SM_DDP_ENABLED = {
+    "smdistributed": {
+        "dataparallel": {
+            "enabled": True,
+            "custom_mpi_options": "options"
+        }
+    }
+}
 
 
 class DummyFramework(Framework):