SagemakerTrainingWithLogTrigger
Astronomer ProvidersSagemakerTrainingWithLogTrigger is fired as deferred class with params to run the task in triggerer.
Access Instructions
Install the Astronomer Providers provider package into your Airflow environment.
Import the module into your DAG file and instantiate it with your desired params.
Parameters
job_namename of the job to check status
instance_countcount of the instance created for running the training job
statusThe status of the training job created.
poke_intervalpolling period in seconds to check for the status
end_timeTime in seconds to wait for a job run to reach a terminal status.
aws_conn_idAWS connection ID for sagemaker
Documentation
SagemakerTrainingWithLogTrigger is fired as deferred class with params to run the task in triggerer.