o
    q¦µiæ  ã                   @   s<   d Z ddlmZ ddlmZ ddlmZ G dd„ deƒZdS )	zParser for lemmatizationé    )ÚCoNLL)Úlemma_lookupé   )ÚParserc                   @   s    e Zd ZdZdd„ Zdd„ ZdS )ÚLemmaParserz'This parser add pos tags for Urdu text.c                 C   s   dS )ÚpassN© )ÚselfÚconfigr   r   úS/home/ubuntu/.local/lib/python3.10/site-packages/urduhack/pipeline/parsers/lemma.pyÚ_set_up   s    zLemmaParser._set_upc                 C   sj   |j D ]/}t|jƒ}t|ƒt|jƒksJ dƒ‚t||jƒD ]\}}|d |jkr1t|tj|d ƒ qq|S )z%Function to normalize|preprocess textz Error in post tagsr   é   )	Ú	sentencesr   ÚtextÚlenÚwordsÚzipÚsetattrr   ÚLEMMA)r	   ÚdocumentÚsentenceÚlemmasÚlemmaÚwordr   r   r   Úparse   s   

€þzLemmaParser.parseN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r   r   
   s    r   N)r   Úurduhack.conllr   Ú%urduhack.models.lemmatizer.lemmatizerr   Úparserr   r   r   r   r   r   Ú<module>   s
   