fix: set _current_job_name in attach() (#808)

Author: jesterhazy

src/sagemaker/estimator.py

@@ -326,6 +326,7 @@ def attach(cls, training_job_name, sagemaker_session=None, model_channel_name='m
         estimator = cls(sagemaker_session=sagemaker_session, **init_params)
         estimator.latest_training_job = _TrainingJob(sagemaker_session=sagemaker_session,
                                                      job_name=init_params['base_job_name'])
+        estimator._current_job_name = estimator.latest_training_job.name
         estimator.latest_training_job.wait()
         return estimator
 

tests/unit/test_estimator.py

@@ -452,6 +452,7 @@ def test_attach_framework(sagemaker_session):
                                                                     return_value=returned_job_description)
 
     framework_estimator = DummyFramework.attach(training_job_name='neo', sagemaker_session=sagemaker_session)
+    assert framework_estimator._current_job_name == 'neo'
     assert framework_estimator.latest_training_job.job_name == 'neo'
     assert framework_estimator.role == 'arn:aws:iam::366:role/SageMakerRole'
     assert framework_estimator.train_instance_count == 1

tests/unit/test_sklearn.py

@@ -324,6 +324,7 @@ def test_attach(sagemaker_session, sklearn_version):
                                                                     return_value=returned_job_description)
 
     estimator = SKLearn.attach(training_job_name='neo', sagemaker_session=sagemaker_session)
+    assert estimator._current_job_name == 'neo'
     assert estimator.latest_training_job.job_name == 'neo'
     assert estimator.py_version == PYTHON_VERSION
     assert estimator.framework_version == sklearn_version