span class="n">dist</span> <span class="o">=</span> <span class="n">model</span><span class="o">.</span><span class="n">worst_distribution</span><span class="p">(</span><span class="n">X</span><span class="p">,</span> <span class="n">y</span><span class="p">)</span>
<span class="gp">&gt;&gt;&gt; </span><span class="n">high_risk_X</span><span class="p">,</span> <span class="n">high_risk_y</span> <span class="o">=</span> <span class="n">dist</span><span class="p">[</span><span class="s2">&quot;sample_pts&quot