Fixed black-check

karan6181 · karan6181 · commit 2f5aed4baf46 · 2021-03-30T09:05:25.000-07:00
diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py
@@ -2607,8 +2607,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
@@ -121,12 +121,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
`@@ -121,12 +121,7 @@`
`121`	`121`	`"mpi": {"enabled": True, "custom_mpi_options": "options", "processes_per_host": 2}`
`122`	`122`	`}`
`123`	`123`	`DISTRIBUTION_SM_DDP_ENABLED = {`
`124`		`- "smdistributed": {`
`125`		`- "dataparallel": {`
`126`		`- "enabled": True,`
`127`		`- "custom_mpi_options": "options"`
`128`		`- }`
`129`		`- }`
	`124`	`+ "smdistributed": {"dataparallel": {"enabled": True, "custom_mpi_options": "options"}}`
`130`	`125`	`}`
`131`	`126`
`132`	`127`