s="n">atom_cluster</span><span class="p">):</span>
        <span class="n">visited</span><span class="p">[</span><span class="n">atom</span><span class="p">]</span> <span class="o">=</span> <span class="kc">True</span>
        <span class="n">new_cluster</span> <span class="o">=</span> <span class="nb">set</span><span class="p">(</span><span class="n">np</span><span class="o">.</span><span class="n">where</span><span class="p">(</span><span class="n">connected_matrix</span><span class="p">[</span><span clas