n><span class="n">cal_dis</span><span class="p">(</span><span class="n">x</span><span class="p">)</span>
        
        <span class="n">idxs</span> <span class="o">=</span> <span class="n">dis</span><span class="o">.</span><span class="n">argsort</span><span class="p">()[:</span><span class="n">budget</span><span class="p">]</span>
        <span class="k">return</span> <span class="n">idxs</span></div></div>
</pre></div>

           </div>
           
          </div>
          <footer>
  

  