[ML][single] Fix SingleShot.input attribute change