o
    …wÖi$  ã                   @   s6   d dl Z d dlmZ G dd„ de jjjƒZdd„ ZdS )é    N)ÚFallbackDatasetc                   @   s   e Zd Zdd„ ZdS )ÚDummyDatasetc                 C   s   |S )N© )ÚselfÚitemr   r   úk/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/tests/collections/common/test_fallback_dataset.pyÚ__getitem__   s   zDummyDataset.__getitem__N)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r      s    r   c                  C   sâ   t ƒ } | d dksJ ‚| d  d u sJ ‚t| ƒ}|d dksJ ‚|d  dks'J ‚|d dks/J ‚|d  dks7J ‚t| ƒ}|d dksCJ ‚|d  dksKJ ‚|d  dksSJ ‚t| ƒ}|d  d u s_J ‚|d dksgJ ‚|d  dksoJ ‚d S )Nr   é   )r   r   )ÚdsetÚfallbackr   r   r   Útest_fallback_dataset   s    r   )ÚtorchÚ%nemo.collections.common.data.fallbackr   ÚutilsÚdataÚDatasetr   r   r   r   r   r   Ú<module>   s   