n>
        <span class="p">}</span></div>

    <span class="nd">@on_rank_0</span>
    <span class="k">def</span> <span class="nf">print_metrics</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">metrics_data</span><span class="p">,</span> <span class="n">dataset</span><span class="p">):</span>
        <span class="sd">&quot;&quot;&quot;Print depth metrics on rank 0 if available&quot;&quot;&quot;</span>
        <span class="k">if</span> <span class="ow">not</sp