o
    à¥µiÄ	  ã                   @   s  d dl mZ d dlmZ d dlmZ dZdZdZej	eej	ej
ejejejgeej	iejeejejejejgeej	iejeejejejejgeej	iejeejejgeejiejeejejgeejiejeejgeejiejeejgeejiiZdd„ ZeeƒZd	d
„ ZdS )é    )Ú
TaskModels)Úregistry)ÚTasksÚ	sub_tasksÚparent_taskÚ
task_modelc                 C   s>   t ƒ }| D ]}| | t D ]}t|t| | t i||< qq|S ©N)ÚdictÚ	SUB_TASKSÚPARENT_TASKÚ
TASK_MODEL)Úforward_indexÚinverted_indexÚindexÚitem© r   úO/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/utils/task_utils.pyÚ_inverted_indexG   s   
þÿr   c                 C   s(   | t v rt |  t t |  t fS | d fS r   )ÚINVERTED_TASKS_LEVELr   r   )Ú	group_keyr   r   r   Úget_task_by_subtask_nameU   s   ÿ
ÿr   N)Úmodelscope.metainfor   Úmodelscope.utilsr   Úmodelscope.utils.constantr   r
   r   r   Útext_classificationÚsentence_similarityÚsentiment_classificationÚsentiment_analysisÚnliÚtoken_classificationÚnamed_entity_recognitionÚword_segmentationÚpart_of_speechÚtext_generationÚtext2text_generationÚinformation_extractionÚrelation_extractionÚ	fill_maskÚtext_rankingÚDEFAULT_TASKS_LEVELr   r   r   r   r   r   r   Ú<module>   s†    û÷üø
üø
þûþûÿüÿüÌ>