Skip to content

Commit 2f5aed4

Browse files
committed
Fixed black-check
1 parent cc9cf32 commit 2f5aed4

File tree

2 files changed

+4
-8
lines changed

2 files changed

+4
-8
lines changed

src/sagemaker/estimator.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2607,8 +2607,9 @@ def _distribution_configuration(self, distribution):
26072607
distribution_config[self.LAUNCH_SM_DDP_ENV_NAME] = smdataparallel_enabled
26082608
distribution_config[self.INSTANCE_TYPE] = self.instance_type
26092609
if smdataparallel_enabled:
2610-
distribution_config[self.SM_DDP_CUSTOM_MPI_OPTIONS] = \
2611-
smdistributed["dataparallel"].get("custom_mpi_options", "")
2610+
distribution_config[self.SM_DDP_CUSTOM_MPI_OPTIONS] = smdistributed[
2611+
"dataparallel"
2612+
].get("custom_mpi_options", "")
26122613

26132614
return distribution_config
26142615

tests/unit/test_estimator.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -121,12 +121,7 @@
121121
"mpi": {"enabled": True, "custom_mpi_options": "options", "processes_per_host": 2}
122122
}
123123
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"}}
130125
}
131126

132127

0 commit comments

Comments
 (0)