
    bis                     .    d dl Zd dlmZ  G d de      Zy)    N)AudioDecoderc                   $     e Zd Zdef fdZ xZS )r   keyc           	         |dk(  r}| j                         j                  j                         j                         }|j                  dkD  r6t        j                  |t        t        |j                  dz
                    S |S |dk(  r| j                  dd      j                  S t        t               d      rt        | 5  |      S t        d      )Narray   )axissampling_rater   __getitem__z>'torchcodec.decoders.AudioDecoder' object is not subscriptable)get_all_samplesdatacpunumpyndimnpmeantuplerangeget_samples_played_in_rangesample_ratehasattrsuperr   	TypeError)selfr   y	__class__s      X/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/datasets/features/_torchcodec.pyr   zAudioDecoder.__getitem__   s    '>$$&++//1779A@A
27715qvvz):#;<QPQQO#33Aq9EEEUWm,7&s++\]]    )__name__
__module____qualname__strr   __classcell__)r   s   @r   r   r      s    	^s 	^ 	^r   r   )r   r   torchcodec.decodersr   _AudioDecoder r   r   <module>r'      s     =
^= 
^r   