n class="s1">&#39;red&#39;</span><span class="p">))</span>

    <span class="c1"># Global shared dataset arguments</span>
    <span class="n">dataset_args</span> <span class="o">=</span> <span class="p">{</span>
        <span class="s1">&#39;back_context&#39;</span><span class="p">:</span> <span class="n">config</span><span class="o">.</span><span class="n">back_context</span><span class="p">,</span>
        <span class="s1">&#39;forward_context&#39;</span><span class="p">:</span> <span class="n">config</sp