U
    -e/                     @   sH   d dl mZmZ ddlmZ ddlmZ ddlmZ G dd deZ	d	S )
    )CallableOptional   )Features)	Generator   )AbstractDatasetInputStreamc                	       sD   e Zd Zdeee eeeee ee	 d fddZ
dd Z  ZS )	GeneratorDatasetInputStreamNF)	generatorfeatures	cache_dirkeep_in_memory	streaming
gen_kwargsnum_procc           	         s<   t  jf |||||d| tf ||||d|| _d S )N)r   r   r   r   r   )r   r   r
   r   )super__init__r   builder)	selfr
   r   r   r   r   r   r   kwargs	__class__ V/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/datasets/io/generator.pyr   	   s"    z$GeneratorDatasetInputStream.__init__c                 C   sV   | j r| jjdd}n<d }d }d }d }| jj||||| jd | jjd|| jd}|S )Ntrain)split)download_configdownload_modeverification_mode	base_pathr   )r   r   Z	in_memory)r   r   Zas_streaming_datasetZdownload_and_preparer   Z
as_datasetr   )r   Zdatasetr   r   r   r   r   r   r   read$   s&      z GeneratorDatasetInputStream.read)NNFFNN)__name__
__module____qualname__r   r   r   strbooldictintr   r    __classcell__r   r   r   r   r	      s          r	   N)
typingr   r    r   Z$packaged_modules.generator.generatorr   abcr   r	   r   r   r   r   <module>   s   