pan class="n">shape</span><span class="p">)</span> <span class="o">==</span> <span class="mi">1</span><span class="p">:</span>
        <span class="n">p</span> <span class="o">=</span> <span class="n">HEG</span><span class="o">.</span><span class="n">vx_lda</span><span class="p">[</span><span class="s2">&quot;spin_polerised&quot;</span><span class="p">]</span>
        <span class="n">q</span> <span class="o">=</span> <span class="n">HEG</span><span class="o">.</span><span class="n">ec_lda</span><span class=