ass="sd">    kwargs : dict</span>
<span class="sd">        Extra parameters for dataset creation</span>

<span class="sd">    Returns</span>
<span class="sd">    -------</span>
<span class="sd">    dataset : Dataset</span>
<span class="sd">        Dataset class for that mode</span>
<span class="sd">    &quot;&quot;&quot;</span>
    <span class="c1"># If no dataset is given, return None</span>
    <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">config</span><span 