U -e{@sfddlZddlmZmZddlmZmZddlmZm Z m Z ddl m Z edd Gd d d e Z dS) N) dataclassfield)ClassVarDict) ClassLabelFeaturesValue) TaskTemplateT)frozenc@seZdZUedddidZeed<edediZ e eed<ed e iZ e eed <dZ eed <d Zeed <d dZeeeefdddZdS)TextClassificationztext-classificationZ$include_in_asdict_even_if_is_defaultT)defaultmetadatatasktextstring input_schemalabels label_schema text_column label_columncCsn|j|krtd|jdt||jts>td|jdt|}|j}||j|d<||jd<|S)NzColumn z is not present in features.z is not a ClassLabel.rr)r ValueError isinstancercopydeepcopyr__dict__)selffeaturesZ task_templaterrc/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/datasets/tasks/text_classification.pyalign_with_featuress    z&TextClassification.align_with_features)returncCs|jd|jdiS)Nrr)rr)rrrr column_mappings z!TextClassification.column_mappingN)__name__ __module__ __qualname__rrstr__annotations__rr rrrrrrr!propertyrr#rrrr r s    r )r dataclassesrrtypingrrrrrr baser r rrrr s