Skip to content

Commit

Permalink
feat(amazon): deprecate SagemakerTrainingWithLogTrigger
Browse files Browse the repository at this point in the history
  • Loading branch information
Lee-W committed Feb 7, 2024
1 parent 8efab14 commit 10a3c8f
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions astronomer/providers/amazon/aws/triggers/sagemaker.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.SageMakerTrainingPrintLogTrigger` 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.
Expand All @@ -209,6 +212,14 @@ def __init__(
end_time: float | None = None,
aws_conn_id: str = "aws_default",
):
warnings.warn(
(
"This module is deprecated and will be removed in 2.0.0."
"Please use `airflow.providers.amazon.aws.hooks.sagemaker.SageMakerTrainingPrintLogTrigger`"
),
DeprecationWarning,
stacklevel=2,
)
super().__init__()
self.job_name = job_name
self.instance_count = instance_count
Expand Down

0 comments on commit 10a3c8f

Please sign in to comment.