U
    -e                     @   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   )Audio
ClassLabelFeatures   )TaskTemplateT)frozenc                   @   s   e Zd ZU edddidZeed< ede 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eeef dddZdS )AudioClassificationzaudio-classificationZ$include_in_asdict_even_if_is_defaultT)defaultmetadatataskaudioinput_schemalabelslabel_schemaaudio_columnlabel_columnc                 C   sn   | j |k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   e/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/datasets/tasks/audio_classificiation.pyalign_with_features   s    



z'AudioClassification.align_with_features)returnc                 C   s   | j d| jdiS )Nr   r   )r   r   )r   r   r   r   column_mapping   s
      z"AudioClassification.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   