o
    h                     @  s   d dl mZ d dlmZmZmZ d dlmZ d dlm	Z
 d dlmZmZmZmZmZmZmZ er>d dlmZmZ d dlmZ e	ddddZe	ddddZ	d d!ddZG dd de
Z	dZdS )"    )annotations)TYPE_CHECKINGCallableoverload)sanitize_pandas_dataframe)DataTransformerRegistry)MaxRowsErrorcheck_data_type
limit_rowssampleto_csvto_json	to_values)DataTypeToValuesReturnType)PluginEnabler.dataNonemax_rowsintreturn(Callable[[DataType], ToValuesReturnType]c                 C     d S N r   r   r   r   H/var/www/vscode/kcb/lib/python3.10/site-packages/altair/vegalite/data.pydefault_data_transformer      r   r   r   c                 C  r   r   r   r   r   r   r   r      r   N  DataType | None=Callable[[DataType], ToValuesReturnType] | ToValuesReturnTypec                   s*   | d u rd fdd}|S t t|  dS )	Nr   r   r   r   c                  s   t |  d} t| S )Nr   )r
   r   )r   r"   r   r   pipe#   s   z&default_data_transformer.<locals>.piper"   )r   r   r   r   )r   r
   )r   r   r#   r   r"   r   r      s   c                   @  s   e Zd ZdddZdS )r   r   r   c                 C  s0   | j }| jdv r| }d|d< | jdi |S )zDisable the MaxRowsError.>   default
vegafusionNr   r   )optionsactivecopyenable)selfr&   r   r   r   disable_max_rows.   s
   
z(DataTransformerRegistry.disable_max_rowsN)r   r   )__name__
__module____qualname__r+   r   r   r   r   r   -   s    r   )
r   r   r	   r   r
   r   r   r   r   r   )..)r   r   r   r   r   r   ).)r   r   r   r   r   r   )Nr   )r   r    r   r   r   r!   )
__future__r   typingr   r   r   altair.utils.corer   altair.utils.datar   _DataTransformerRegistryr   r	   r
   r   r   r   r   r   r   altair.utils.plugin_registryr   r   __all__r   r   r   r   <module>   s$    $

