Fix flake style errors

vishwakaria · vishwakaria · commit fcc6235867e4 · 2022-07-12T01:32:31.000Z
diff --git a/src/sagemaker_pytorch_container/training.py b/src/sagemaker_pytorch_container/training.py
@@ -52,7 +52,7 @@ def train(training_environment):
     _set_distributed_environment(training_environment.hosts)
 
     mpi_enabled = training_environment.additional_framework_parameters.get(LAUNCH_MPI_ENV_NAME)
-    
+
     pytorch_ddp_enabled = training_environment.additional_framework_parameters.get(
         LAUNCH_PYTORCH_DDP_ENV_NAME, False
     )
diff --git a/test/unit/test_train.py b/test/unit/test_train.py
@@ -87,8 +87,8 @@ def test_train_smdataparallel(run_module, training_env):
         capture_error=True,
         runner_type=runner.SMDataParallelRunnerType,
     )
-    
-    
+
+
 @patch("sagemaker_training.entry_point.run")
 @patch('socket.gethostbyname', MagicMock())
 def test_train_pytorch_ddp(run_module, training_env):

Original file line number	Diff line number	Diff line change
`@@ -52,7 +52,7 @@ def train(training_environment):`
`52`	`52`	`_set_distributed_environment(training_environment.hosts)`
`53`	`53`
`54`	`54`	`mpi_enabled = training_environment.additional_framework_parameters.get(LAUNCH_MPI_ENV_NAME)`
`55`		`-`
	`55`	`+`
`56`	`56`	`pytorch_ddp_enabled = training_environment.additional_framework_parameters.get(`
`57`	`57`	`LAUNCH_PYTORCH_DDP_ENV_NAME, False`
`58`	`58`	`)`