diff --git a/src/sagemaker/local/local_session.py b/src/sagemaker/local/local_session.py index 98da459bbc..4c2f02cad1 100644 --- a/src/sagemaker/local/local_session.py +++ b/src/sagemaker/local/local_session.py @@ -571,6 +571,21 @@ def logs_for_job(self, job_name, wait=False, poll=5, log_type="All"): # on local mode. pass # pylint: disable=unnecessary-pass + def logs_for_processing_job(self, job_name, wait=False, poll=10): + """A no-op method meant to override the sagemaker client. + + Args: + job_name: + wait: (Default value = False) + poll: (Default value = 10) + + Returns: + + """ + # override logs_for_job() as it doesn't need to perform any action + # on local mode. + pass # pylint: disable=unnecessary-pass + class file_input(object): """Amazon SageMaker channel configuration for FILE data sources, used in local mode.""" diff --git a/tests/unit/test_local_session.py b/tests/unit/test_local_session.py index 13875fe292..6d64c73849 100644 --- a/tests/unit/test_local_session.py +++ b/tests/unit/test_local_session.py @@ -551,6 +551,20 @@ def test_describe_transform_job_does_not_exist(LocalSession, _LocalTransformJob) local_sagemaker_client.describe_transform_job("transform-job-does-not-exist") +@patch("sagemaker.local.image._SageMakerContainer.process") +@patch("sagemaker.local.local_session.LocalSession") +def test_logs_for_job(process, LocalSession): + local_job_logs = LocalSession.logs_for_job("my-processing-job") + assert local_job_logs is not None + + +@patch("sagemaker.local.image._SageMakerContainer.process") +@patch("sagemaker.local.local_session.LocalSession") +def test_logs_for_processing_job(process, LocalSession): + local_processing_job_logs = LocalSession.logs_for_processing_job("my-processing-job") + assert local_processing_job_logs is not None + + @patch("sagemaker.local.local_session.LocalSession") def test_describe_endpoint_config(LocalSession): local_sagemaker_client = sagemaker.local.local_session.LocalSagemakerClient()