class="s2">&quot;Q_a&quot;</span><span class="p">]</span>
        <span class="n">Q_b</span> <span class="o">=</span> <span class="n">kwargs</span><span class="p">[</span><span class="s2">&quot;Q_b&quot;</span><span class="p">]</span>

        <span class="n">a_star</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">argmax</span><span class="p">(</span><span class="n">Q_a</span><span class="p">[</span><span class="nb">tuple</span><span class="p">(</span><span cl