fix: Change Estimator.logs() to use latest_training_job.name (#2059)

Co-authored-by: ChoiByungWook <[email protected]>

Author: philschmid

src/sagemaker/estimator.py

@@ -820,7 +820,7 @@ def logs(self):
         If the output is a tty or a Jupyter cell, it will be color-coded based
         on which instance the log entry is from.
         """
-        self.sagemaker_session.logs_for_job(self.latest_training_job, wait=True)
+        self.sagemaker_session.logs_for_job(self.latest_training_job.name, wait=True)
 
     def deploy(
         self,

tests/unit/test_estimator.py

@@ -1434,7 +1434,7 @@ def test_attach_no_logs(sagemaker_session, training_job_description):
 def test_logs(sagemaker_session, training_job_description):
     estimator = Estimator.attach(training_job_name="job", sagemaker_session=sagemaker_session)
     estimator.logs()
-    sagemaker_session.logs_for_job.assert_called_with(estimator.latest_training_job, wait=True)
+    sagemaker_session.logs_for_job.assert_called_with(estimator.latest_training_job.name, wait=True)
 
 
 def test_attach_without_hyperparameters(sagemaker_session, training_job_description):