_data</span><span class="p">[</span><span class="s2">&quot;step_count&quot;</span><span class="p">]</span> <span class="o">+=</span> <span class="mi">1</span></div>

    
<div class="viewcode-block" id="CustomCallbacks.on_episode_end">
<a class="viewcode-back" href="../../code/utils.html#utils.rllib_callbacks.CustomCallbacks.on_episode_end">[docs]</a>
    <span class="k">def</span> <span class="nf">on_episode_end</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class=