Skip to content

Commit 7751e62

Browse files
author
Chuyang Deng
committed
update unit test to test not raising error when wait is False logs is True
1 parent 6e5a794 commit 7751e62

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/sagemaker/automl/automl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def fit(self, inputs=None, wait=True, logs=True, job_name=None):
7979
be uploaded to an S3 location.
8080
wait (bool): Whether the call should wait until the job completes (default: True).
8181
logs (bool): Whether to show the logs produced by the job. Only meaningful when wait
82-
is True (default: True). if `wait` is False, `log` will be set to False as well.
82+
is True (default: True). if `wait` is False, `logs` will be set to False as well.
8383
job_name (str): Training job name. If not specified, the estimator generates
8484
a default job name, based on the training image name and current timestamp.
8585
"""

tests/unit/sagemaker/automl/test_auto_ml.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,7 @@ def test_auto_ml_additional_optional_params(sagemaker_session):
248248
tags=TAGS,
249249
)
250250
inputs = DEFAULT_S3_INPUT_DATA
251-
auto_ml.fit(inputs, job_name=JOB_NAME)
251+
auto_ml.fit(inputs, job_name=JOB_NAME, wait=False, logs=True)
252252
sagemaker_session.auto_ml.assert_called_once()
253253
_, args = sagemaker_session.auto_ml.call_args
254254

0 commit comments

Comments
 (0)