o
    0j5                     @   s@   d dl ZddlmZ ddlmZ ejdddG dd dZdS )	    N   )	benchmark)	CSVReaderT)nameZis_read_operationc                       s,   e Zd Z fddZdd Zdd Z  ZS )ReadTSc                    s   t    tdd| _d S )Npandas)backend)super__init__r   _readerself	__class__ j/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/common/reader/ts_reader.pyr
      s   
zReadTS.__init__c                    s    fdd|D S )applyc                    s   g | ]}  |qS r   )read).0tsr   r   r   
<listcomp>   s    z#ReadTS.__call__.<locals>.<listcomp>r   )r   Zts_listr   r   r   __call__   s   zReadTS.__call__c                 C   sV   t |tjr|S t |tr | j|}|d u rtd| |S tdt|j	 d)NzTS read Error: zReadTS only supports the following types:
1. str, indicating a CSV file path or a directory containing CSV files.
2. pandas.DataFrame.
However, got type: .)

isinstancepdZ	DataFramestrr   r   	Exception	TypeErrortype__name__)r   r   Zts_datar   r   r   r       s   
zReadTS.read)r   
__module____qualname__r
   r   r   __classcell__r   r   r   r   r      s    r   )r   r   Zutils.benchmarkr   Zutils.ior   Ztimeit_with_optionsr   r   r   r   r   <module>   s
   