diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py index 999f5d10f0..b1f8d45241 100644 --- a/src/sagemaker/estimator.py +++ b/src/sagemaker/estimator.py @@ -2116,9 +2116,9 @@ def __init__( https://docs.aws.amazon.com/sagemaker/latest/dg/model-managed-spot-training.html (default: ``False``). max_wait (int): Timeout in seconds waiting for spot training - instances (default: None). After this amount of time Amazon - SageMaker will stop waiting for Spot instances to become - available (default: None). + job (default: None). After this amount of time Amazon + SageMaker will stop waiting for managed spot training job to + complete (default: None). checkpoint_s3_uri (str): The S3 URI in which to persist checkpoints that the algorithm persists (if any) during training. (default: None).