feat(amazon): deprecate SagemakerTrainingWithLogTrigger

astronomer · Jan 30, 2024 · 534951a · 534951a
1 parent 32599fe
commit 534951a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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.