Hi,

I had a job that got stuck on yarn due to https://issues.apache.org/jira/browse/SPARK-6954 
It never exited properly. 

Is there a way to set a timeout for a stage or all stages?