an class="p">:</span>
        <span class="n">key_tokens</span> <span class="o">=</span> <span class="n">query_tokens</span>

    <span class="kn">from</span> <span class="nn">inspectus.attention_viz</span> <span class="kn">import</span> <span class="n">parse_attn</span><span class="p">,</span> <span class="n">attention_chart</span><span class="p">,</span> <span class="n">parse_colors</span>

    <span class="n">attn</span><span class="p">,</span> <span class="n">dimensions</span> <span class="o">=</span> <