ss="n">reset</span><span class="p">()</span>
            <span class="k">else</span><span class="p">:</span>
                <span class="k">if</span> <span class="n">np</span><span class="o">.</span><span class="n">all</span><span class="p">(</span>
                    <span class="n">done</span>
                <span class="p">):</span>  <span class="c1"># if done, save the original obs, state, and available actions in info, and then reset</span>
                    <span class="n">info</span><span class=