class="o">.</span><span class="n">device</span><span class="p">(</span><span class="s2">&quot;cpu&quot;</span><span class="p">)):</span>
        <span class="k">assert</span> <span class="p">(</span>
            <span class="n">act_space</span><span class="o">.</span><span class="vm">__class__</span><span class="o">.</span><span class="vm">__name__</span> <span class="o">==</span> <span class="s2">&quot;Box&quot;</span>
        <span class="p">),</span> <span class="sa">f</span><span class="s2">&quot;only c