Skip to content

Commit 96d6b3a

Browse files
committed
auto-generating sphinx docs
1 parent a77a026 commit 96d6b3a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

main/_modules/torchtune/utils/_logging.html

+3
Original file line numberDiff line numberDiff line change
@@ -515,6 +515,9 @@ <h1>Source code for torchtune.utils._logging</h1><div class="highlight"><pre>
515515

516516
<span class="nd">@lru_cache</span><span class="p">(</span><span class="n">maxsize</span><span class="o">=</span><span class="mi">1</span><span class="p">)</span>
517517
<span class="k">def</span><span class="w"> </span><span class="nf">warn</span><span class="p">(</span><span class="n">obj</span><span class="p">):</span>
518+
<span class="n">rank</span> <span class="o">=</span> <span class="n">dist</span><span class="o">.</span><span class="n">get_rank</span><span class="p">()</span> <span class="k">if</span> <span class="n">dist</span><span class="o">.</span><span class="n">is_available</span><span class="p">()</span> <span class="ow">and</span> <span class="n">dist</span><span class="o">.</span><span class="n">is_initialized</span><span class="p">()</span> <span class="k">else</span> <span class="mi">0</span>
519+
<span class="k">if</span> <span class="n">rank</span> <span class="o">!=</span> <span class="mi">0</span><span class="p">:</span>
520+
<span class="k">return</span>
518521
<span class="n">warnings</span><span class="o">.</span><span class="n">warn</span><span class="p">(</span>
519522
<span class="sa">f</span><span class="s2">&quot;</span><span class="si">{</span><span class="n">obj</span><span class="o">.</span><span class="vm">__name__</span><span class="si">}</span><span class="s2"> is deprecated and will be removed in future versions. &quot;</span>
520523
<span class="o">+</span> <span class="n">msg</span><span class="p">,</span>

0 commit comments

Comments
 (0)