add warning

Author: Chuyang Deng

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
@@ -83,8 +86,11 @@ def fit(self, inputs=None, wait=True, logs=True, job_name=None):
             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 not wait:
+        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

@@ -228,7 +228,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,
@@ -282,6 +282,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)