s="n">ndarray</span><span class="p">,</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">],</span> <span class="o">...</span><span class="p">]:</span>
    <span class="k">if</span> <span class="n">estimation_method</span> <span class="o">==</span> <span class="s1">&#39;prediction&#39;</span><span class="p">:</span>
        <span class="c1"># make sure types are correct for:</span>
        <span class="k">if</span> <span class="nb">isinstance</span><span c