
    g                        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       Ze	 d	 	 	 	 	 dd	       Z	 d	 	 	 	 	 ddZ G d de
      Z	dZy
)    )annotations)TYPE_CHECKINGCallableoverload)sanitize_pandas_dataframe)DataTransformerRegistry)MaxRowsErrorcheck_data_type
limit_rowssampleto_csvto_json	to_values)DataTypeToValuesReturnType)PluginEnablerc                     y N datamax_rowss     I/var/www/openai/venv/lib/python3.12/site-packages/altair/vegalite/data.pydefault_data_transformerr      s     03    c                     y r   r   r   s     r   r   r      s     r   Nc                D    | dfd}|S t        t        |             S )Nc               4    t        |       } t        |       S )Nr   )r   r   r   s    r   pipez&default_data_transformer.<locals>.pipe#   s    dX6DT?"r   r   )r   r   returnr   )r   r   )r   r   r    s    ` r   r   r      s*     |	#  D8<==r   c                      e Zd ZddZy)r   c                    | j                   }| j                  dv r|j                         }d|d<    | j                  di |S )zDisable the MaxRowsError.>   default
vegafusionNr   r   )optionsactivecopyenable)selfr&   s     r   disable_max_rowsz(DataTransformerRegistry.disable_max_rows.   sA    ,,;;33llnG"&GJt{{%W%%r   N)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   Noner   intr!   z(Callable[[DataType], ToValuesReturnType]).)r   r   r   r0   r!   r   )Ni  )r   zDataType | Noner   r0   r!   z=Callable[[DataType], ToValuesReturnType] | ToValuesReturnType)
__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   <module>r8      s    " 4 4 7 Q   >: 
&)3
3 #3-3 
3 
$'
! 
 37>
>,/>B>&6 &r   