diff --git a/astronomer/providers/amazon/aws/triggers/sagemaker.py b/astronomer/providers/amazon/aws/triggers/sagemaker.py index f311d3a42..fa928c650 100644 --- a/astronomer/providers/amazon/aws/triggers/sagemaker.py +++ b/astronomer/providers/amazon/aws/triggers/sagemaker.py @@ -189,6 +189,9 @@ class SagemakerTrainingWithLogTrigger(BaseTrigger): """ SagemakerTrainingWithLogTrigger is fired as deferred class with params to run the task in triggerer. + This class is deprecated and will be removed in 2.0.0. + Use :class: `~airflow.providers.amazon.aws.triggers.sagemaker.SageMakerTrigger` instead + :param job_name: name of the job to check status :param instance_count: count of the instance created for running the training job :param status: The status of the training job created.