o
    wZh                     @   sJ   d dl mZ d dlmZ d dlmZ d dlmZ dgZ	G dd deZ
dS )    )Iterator)Any)dataframe_wrapper)	DataChunkDataChunkDFc                   @   s0   e Zd ZdZdee fddZdefddZdS )r   zuDataChunkDF iterating over individual items inside of DataFrame containers, to access DataFrames user `raw_iterator`.returnc                 c   s"    | j D ]
}t|E d H  qd S )N)items
df_wrapperZiterate)selfdf r   ^/var/www/auris/lib/python3.10/site-packages/torch/utils/data/datapipes/dataframe/structures.py__iter__   s   
zDataChunkDF.__iter__c                 C   s"   d}| j D ]	}|t|7 }q|S )Nr   )r   r	   Zget_len)r
   Z	total_lenr   r   r   r   __len__   s   
zDataChunkDF.__len__N)	__name__
__module____qualname____doc__r   r   r   intr   r   r   r   r   r      s    N)collections.abcr   typingr   Z$torch.utils.data.datapipes.dataframer   r	   Z#torch.utils.data.datapipes.datapiper   __all__r   r   r   r   r   <module>   s    