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)