    <span class="n">total_tasks_dropped</span> <span class="o">=</span> <span class="mi">0</span>
            <span class="n">total_water_usage</span> <span class="o">=</span> <span class="mi">0</span>
        
        <span class="n">episode</span><span class="o">.</span><span class="n">custom_metrics</span><span class="p">[</span><span class="s2">&quot;average_total_energy_with_battery&quot;</span><span class="p">]</span> <span class="o">=</span> <span class="n">average_net_energy</span>
        <span cla