black, pylint

Author: laurenyu

src/sagemaker/estimator.py

@@ -329,10 +329,12 @@ def prepare_workflow_for_training(self, job_name=None):
         self._prepare_for_training(job_name=job_name)
 
     def _ensure_base_job_name(self):
+        """Set ``self.base_job_name`` if it is not set already."""
         # honor supplied base_job_name or generate it
         if self.base_job_name:
             return
-        elif isinstance(self, sagemaker.algorithm.AlgorithmEstimator):
+
+        if isinstance(self, sagemaker.algorithm.AlgorithmEstimator):
             self.base_job_name = self.algorithm_arn.split("/")[-1]  # pylint: disable=no-member
         else:
             self.base_job_name = base_name_from_image(self.train_image())

tests/integ/test_mxnet_train.py

@@ -71,10 +71,7 @@ def test_attach_deploy(mxnet_training_job, sagemaker_session, cpu_instance_type)
 
 
 def test_deploy_estimator_with_different_instance_types(
-    mxnet_training_job,
-    sagemaker_session,
-    cpu_instance_type,
-    alternative_cpu_instance_type,
+    mxnet_training_job, sagemaker_session, cpu_instance_type, alternative_cpu_instance_type,
 ):
     def _deploy_estimator_and_assert_instance_type(estimator, instance_type):
         # don't use timeout_and_delete_endpoint_by_name because this tests if
@@ -100,6 +97,7 @@ def _deploy_estimator_and_assert_instance_type(estimator, instance_type):
         return (model_name, endpoint_name, config_name)
 
     estimator = MXNet.attach(mxnet_training_job, sagemaker_session)
+    estimator.base_job_name = "test-mxnet-deploy-twice"
 
     old_model_name, old_endpoint_name, old_config_name = _deploy_estimator_and_assert_instance_type(
         estimator, cpu_instance_type