o
    ?e1	                     @   sR   d Z ddlmZ ddlmZ ddlmZ ddlmZ	 dd Z
G dd	 d	ejZd
S )z,The implementation of `tf.data.Dataset.zip`.    )dataset_ops)nest)gen_dataset_ops)datac                 C   s
   t | |S N)_ZipDataset)datasetsname r
   b/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/data/ops/zip_op.py_zip   s   
r   c                       s6   e Zd ZdZd	 fdd	Zdd Zedd Z  ZS )
r   z*A `Dataset` that zips its inputs together.Nc                    s   t |D ]}t|tjs t|trtdtdt| dq|| _t 	| jdd t | jD | _
|| _tjdd t | jD fi | j}t | dS )z See `Dataset.zip()` for details.zInvalid input to `zip`. Inputs are expected to be (nested) structures of `tf.data.Dataset` objects. Python `list` is not supported and you should use `tuple` instead.zInvalid input to `zip`. Inputs are expected to be (nested) structures of `tf.data.Dataset` objects but encountered object of type .c                 S      g | ]}|j qS r
   )element_spec.0dsr
   r
   r   
<listcomp>0       z(_ZipDataset.__init__.<locals>.<listcomp>c                 S   r   r
   )Z_variant_tensorr   r
   r
   r   r   4   r   N)r   flatten
isinstance
data_types	DatasetV2list	TypeErrortype	_datasetsZpack_sequence_as
_structure_namer   Zzip_datasetZ_common_argssuper__init__)selfr   r	   r   Zvariant_tensor	__class__r
   r   r       s.   
z_ZipDataset.__init__c                 C   s   t | jS r   )r   r   r   r!   r
   r
   r   _inputs9   s   z_ZipDataset._inputsc                 C   s   | j S r   )r   r$   r
   r
   r   r   <   s   z_ZipDataset.element_specr   )	__name__
__module____qualname____doc__r    r%   propertyr   __classcell__r
   r
   r"   r   r      s    r   N)r)   Ztensorflow.python.data.opsr   Ztensorflow.python.data.utilr   Ztensorflow.python.opsr   Ztensorflow.python.typesr   r   r   r   r   r
   r
   r
   r   <module>   s   