o
    TZh{                     @   sf   d dl Z d dlmZmZ d dlmZmZ ddlmZm	Z	m
Z
 ddlmZ edd	G d
d deZdS )    N)	dataclassfield)ClassVarDict   )
ClassLabelFeaturesValue   )TaskTemplateT)frozenc                   @   s   e Zd ZU edddidZeed< edediZ	e
e ed< ed	eiZe
e ed
< dZeed< d	Zeed< dd Zedeeef fddZdS )TextClassificationztext-classificationZ$include_in_asdict_even_if_is_defaultT)defaultmetadatatasktextstringinput_schemalabelslabel_schematext_columnlabel_columnc                 C   sn   | j |vrtd| j  dt|| j  tstd| j  dt| }| j }|| j  |d< ||jd< |S )NzColumn z is not present in features.z is not a ClassLabel.r   r   )r   
ValueError
isinstancer   copydeepcopyr   __dict__)selffeaturesZtask_templater    r   Y/var/www/html/lang_env/lib/python3.10/site-packages/datasets/tasks/text_classification.pyalign_with_features   s   



z&TextClassification.align_with_featuresreturnc                 C   s   | j d| jdiS )Nr   r   )r   r   )r   r   r   r    column_mapping   s   z!TextClassification.column_mappingN)__name__
__module____qualname__r   r   str__annotations__r   r	   r   r   r   r   r   r   r!   propertyr   r#   r   r   r   r    r   	   s   
 r   )r   dataclassesr   r   typingr   r   r   r   r   r	   baser   r   r   r   r   r    <module>   s    