><span class="p">)</span>


<span class="k">class</span> <span class="nc">_MultiDeviceSafeModeContextManager</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span>
<span class="w">    </span><span class="sd">&quot;&quot;&quot;Helper class used in conjunction with `set_multi_device_safe_mode`</span>

<span class="sd">    Used to enable `set_multi_device_safe_mode` as a dual-purpose context manager</span>
<span class="sd">    &quot;&quot;&quot;</span>

    <span class="k">def</s