 class="kn">import</span> <span class="n">to_hetero</span>
    <span class="kn">from</span> <span class="nn">segger.training.train</span> <span class="kn">import</span> <span class="n">LitSegger</span>
    <span class="kn">from</span> <span class="nn">lightning</span> <span class="kn">import</span> <span class="n">Trainer</span>
    <span class="n">logging</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">&quot;Done.&quot;</span><span class="p">)</span>

    <