span></span><span class="kn">from</span> <span class="nn">m3tl.test_base</span> <span class="kn">import</span> <span class="n">TestBase</span>
<span class="kn">from</span> <span class="nn">m3tl.special_tokens</span> <span class="kn">import</span> <span class="n">TRAIN</span>
<span class="kn">from</span> <span class="nn">m3tl.utils</span> <span class="kn">import</span> <span class="n">create_dict_from_nested_model</span>

<span class="n">tb</span> <span class="o">=</span> <span class="n">TestBase</span><span