o 8wis@s(ddlZddlmZGdddeZdS)N) AudioDecodercs"eZdZdeffdd ZZS)rkeycsz|dkr"|j}|jdkr tj|tt|jddS|S|dkr-| ddj St t dr9t |Std)Narray)axis sampling_rater __getitem__z>'torchcodec.decoders.AudioDecoder' object is not subscriptable)get_all_samplesdatacpunumpyndimnpmeantuplerangeget_samples_played_in_range sample_ratehasattrsuperr TypeError)selfry __class__Z/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/datasets/features/_torchcodec.pyrs*  zAudioDecoder.__getitem__)__name__ __module__ __qualname__strr __classcell__rrrrrsr)r rtorchcodec.decodersr _AudioDecoderrrrrs