I added two static variables under class SageMakerHook.

non_terminal_states = {'InProgress', 'Stopping', 'Stopped'}
failed_states = {'Failed'}

And all codes in Hook and Sensors will just use these two static variables. If 
in the future we have an API change for all of training/tuning/inference, we 
only need to change this part. But if we changed API specifically for part of 
them, we still need to go back to previous design. 

For now I would say API should be stable at this part, so used static variables.

[ Full content available at: 
https://github.com/apache/incubator-airflow/pull/3767 ]
This message was relayed via gitbox.apache.org for [email protected]

Reply via email to