o
    Si                     @   sJ   d dl mZmZ d dlZd dlmZ d dlmZ G dd dejj	j
ZdS )    )AnyDictN)CutSetcollate_videoc                   @   s*   e Zd ZdZdedeeef fddZdS )UnsupervisedAudioVideoDataseta  
    A basic dataset that loads, pads, collates, and returns video and audio tensors.

    Returns:

    .. code-block::

        {
            'video': (B x NumFrames x Color x Height x Width) uint8 tensor
            'video_lens': (B, ) int32 tensor
            'audio': (B x NumChannels x NumSamples) float32 tensor
            'audio_lens': (B, ) int32 tensor
            'cuts': CutSet of length B
        }
    cutsreturnc                 C   s&   t |dd\}}}}}|||||dS )NT)fault_tolerant)r   video
video_lensaudio
audio_lensr   )selfr   r   r   r   r    r   H/home/ubuntu/.local/lib/python3.10/site-packages/lhotse/dataset/video.py__getitem__   s   z)UnsupervisedAudioVideoDataset.__getitem__N)	__name__
__module____qualname____doc__r   r   strr   r   r   r   r   r   r   	   s    r   )typingr   r   torchlhotser   lhotse.dataset.collationr   utilsdataDatasetr   r   r   r   r   <module>   s
    