def contrast_strf(modelspec, chans, ax):
try:
strf_heatmap(modelspec, title='Contrast STRF', chans=chans,
wc_idx=1, fir_idx=1, ax=ax)
except IndexError:
strf_heatmap(modelspec, title='Contrast STRF (Fixed A.V.)',
chans=chans, wc_idx=0, fir_idx=0,
absolute_value=True, ax=ax)
def my_strf(idx, ax):
strf_heatmap(modelspecs[idx], ax=ax)