Update wp_scalogram.py