o
    0h                     @   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_wrapperiterate)selfdf r   c/var/www/vscode/kcb/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	   get_len)r   	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   $torch.utils.data.datapipes.dataframer   r	   #torch.utils.data.datapipes.datapiper   __all__r   r   r   r   r   <module>   s    