Fixed black-check

karan6181 · karan6181 · commit a5e0822b6524 · 2021-04-13T09:05:51.000-07:00
diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py
@@ -2631,8 +2631,9 @@ 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
 
diff --git a/tests/unit/test_estimator.py b/tests/unit/test_estimator.py
@@ -122,12 +122,7 @@
     "mpi": {"enabled": True, "custom_mpi_options": "options", "processes_per_host": 2}
 }
 DISTRIBUTION_SM_DDP_ENABLED = {
-    "smdistributed": {
-        "dataparallel": {
-            "enabled": True,
-            "custom_mpi_options": "options"
-        }
-    }
+    "smdistributed": {"dataparallel": {"enabled": True, "custom_mpi_options": "options"}}
 }
 
 

Original file line number	Diff line number	Diff line change
`@@ -122,12 +122,7 @@`
`122`	`122`	`"mpi": {"enabled": True, "custom_mpi_options": "options", "processes_per_host": 2}`
`123`	`123`	`}`
`124`	`124`	`DISTRIBUTION_SM_DDP_ENABLED = {`
`125`		`- "smdistributed": {`
`126`		`- "dataparallel": {`
`127`		`- "enabled": True,`
`128`		`- "custom_mpi_options": "options"`
`129`		`- }`
`130`		`- }`
	`125`	`+ "smdistributed": {"dataparallel": {"enabled": True, "custom_mpi_options": "options"}}`
`131`	`126`	`}`
`132`	`127`
`133`	`128`