fix: set logs to False if wait is False

src/sagemaker/automl/automl.py

@@ -13,6 +13,7 @@
 """A class for SageMaker AutoML Jobs."""
 from __future__ import absolute_import
 
+import logging
 from six import string_types
 
 from sagemaker import Model, PipelineModel
@@ -21,6 +22,8 @@
 from sagemaker.session import Session
 from sagemaker.utils import name_from_base
 
+logger = logging.getLogger("sagemaker")
+
 
 class AutoML(object):
     """A class for creating and interacting with SageMaker AutoML jobs
@@ -78,16 +81,14 @@ def fit(self, inputs=None, wait=True, logs=True, job_name=None):
                 is stored. Or an AutoMLInput object. If a local path is provided, the dataset will
                 be uploaded to an S3 location.
             wait (bool): Whether the call should wait until the job completes (default: True).
-            logs (bool): Whether to show the logs produced by the job.
-                Only meaningful when wait is True (default: True).
+            logs (bool): Whether to show the logs produced by the job. Only meaningful when wait
+                is True (default: True). if `wait` is False, `logs` will be set to False as well.
             job_name (str): Training job name. If not specified, the estimator generates
                 a default job name, based on the training image name and current timestamp.
         """
-        if logs and not wait:
-            raise ValueError(
-                """Logs can only be shown if wait is set to True.
-                Please either set wait to True or set logs to False."""
-            )
+        if not wait and logs:
+            logs = False
+            logger.warning("logs will be set to False. logs is only meaningful when wait is True.")
 
         # upload data for users if provided local path
         # validations are done in _Job._format_inputs_to_input_config

tests/unit/sagemaker/automl/test_auto_ml.py

@@ -294,7 +294,7 @@ def test_auto_ml_only_one_of_problem_type_and_job_objective_provided(sagemaker_s
         )
 
 
-def test_auto_ml_additional_optional_params(sagemaker_session):
+def test_auto_ml_additional_optional_params(sagemaker_session, caplog):
     auto_ml = AutoML(
         role=ROLE,
         target_attribute_name=TARGET_ATTRIBUTE_NAME,
@@ -314,7 +314,7 @@ def test_auto_ml_additional_optional_params(sagemaker_session):
         tags=TAGS,
     )
     inputs = DEFAULT_S3_INPUT_DATA
-    auto_ml.fit(inputs, job_name=JOB_NAME)
+    auto_ml.fit(inputs, job_name=JOB_NAME, wait=False, logs=True)
     sagemaker_session.auto_ml.assert_called_once()
     _, args = sagemaker_session.auto_ml.call_args
 
@@ -348,6 +348,7 @@ def test_auto_ml_additional_optional_params(sagemaker_session):
         "generate_candidate_definitions_only": GENERATE_CANDIDATE_DEFINITIONS_ONLY,
         "tags": TAGS,
     }
+    assert "logs will be set to False. logs is only meaningful when wait is True." in caplog.text
 
 
 @patch("time.strftime", return_value=TIMESTAMP)