From 534951a54afdf613aebfdefaa5f4e6fff889392d Mon Sep 17 00:00:00 2001 From: Wei Lee Date: Tue, 30 Jan 2024 17:49:04 +0800 Subject: [PATCH] feat(amazon): deprecate SagemakerTrainingWithLogTrigger --- astronomer/providers/amazon/aws/triggers/sagemaker.py | 3 +++ 1 file changed, 3 insertions(+) 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.