
    vjU                     F    d dl mZ  G d d          Z G d d          ZdS )    )defaultdictc                       e Zd Zd ZdS )TreeNodec                 8    t          t                    | _        d S N)r   r   child)selfs    e/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/modelscope/utils/trie.py__init__zTreeNode.__init__   s     **


    N)__name__
__module____qualname__r    r   r
   r   r      s#        + + + + +r   r   c                        e Zd Zd Zd Zd ZdS )Triec                 :    t                      | _        || _        d S r   )r   rooteos)r	   r   s     r
   r   zTrie.__init__   s    JJ	r   c                 8    | j         }|D ]}|j        |         }d S r   )r   r   r	   wordcurcs       r
   insertzTrie.insert   s/    i 	 	A)A,CC	 	r   c                     | j         }|D ](}|j                            |          }|
| j        gc S )t	          |j                                                  S r   )r   r   getr   listkeysr   s       r
   get_next_layerzTrie.get_next_layer   s_    i 	" 	"A)--""C{z!!! CINN$$%%%r   N)r   r   r   r   r   r    r   r   r
   r   r      sA            
& & & & &r   r   N)collectionsr   r   r   r   r   r
   <module>r"      ss    $ # # # # #+ + + + + + + +& & & & & & & & & &r   