an><span class="o">.</span><span class="n">forward_context</span><span class="p">,</span>
        <span class="s1">&#39;data_transform&#39;</span><span class="p">:</span> <span class="n">get_transforms</span><span class="p">(</span><span class="n">mode</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
    <span class="p">}</span>

    <span class="c1"># Loop over all datasets</span>
    <span class="n">datasets</span> <span class="o">=</span> <span