Update scale_amp.py