+
    QV-j¸  ã                   ó:  € ^ RI t ^ RIHt ^ RIHtHtHt ^ RIt^ RI	t
^ RIt^ RIt^ RIHt ^ RIHt ^ RIHt ]'       d	   ^ RIt^ RIt]P*                  P,                  P/                  ]4      t] ! R R]P4                  4      4       t ! R	 R
]P8                  4      tR# )é    N)Ú	dataclass)ÚTYPE_CHECKINGÚOptionalÚUnion)ÚKey©Úrequire_storage_cast)Ú
table_castc                   ó’   a a€ ] tR t^t oRtRtRtRtRtRt	Rt
RtRtRtV 3R ltRV3R lV 3R lllt]R 4       tV3R	 ltR
tVtV ;t# )Ú	SqlConfigzBuilderConfig for SQL.NTi'  c                óˆ   <€ \         SV `  4        V P                  f   \        R4      hV P                  f   \        R4      hR # )Nzsql must be specifiedzcon must be specified)ÚsuperÚ__post_init__ÚsqlÚ
ValueErrorÚcon)ÚselfÚ	__class__s   &€Úr/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/datasets/packaged_modules/sql/sql.pyr   ÚSqlConfig.__post_init__&   s?   ø€ Ü‰ÑÔØ�8‰8ÒÜÐ4Ó5Ð5Ø�8‰8ÒÜÐ4Ó5Ð5ñ ó    c                óP   <€ V ^8„  d   QhRS[ RS[S[P                  ,          RS[/# )é   Úconfig_kwargsÚcustom_featuresÚreturn)Údictr   ÚdatasetsÚFeaturesÚstr)ÚformatÚ__classdict__s   "€r   Ú__annotate__ÚSqlConfig.__annotate__-   s9   ø€ ÷ !Xñ !Xáð!Xñ "¡(×"3Ñ"3Õ4ð!Xñ 
ñ	!Xr   c                ó  <€ VP                  4       pVR ,          p\        V\        4      '       gó   \        P                  P
                  '       d¹   R\        P                  9   d¤   ^ RIp\        W4P                  P                  4      '       da   VP                  ! VR,          P                  R4      ^ ,          R,           4      p\        VP                  VP                  R7      4      pWaR &   M4\        R\!        V4       RV 24      h\        R\!        V4       RV 24      hVR,          p\        V\        4      '       g1   \#        V4      VR&   \$        P'                  R\!        V4       R	24       \(        SV `U  WR
7      # )r   Ú
sqlalchemyNr   z://)ÚdialectzKSupported types for 'sql' are string and sqlalchemy.sql.Selectable but got z: zSQL connection 'con' of type zU couldn't be hashed properly. To enable hashing, specify 'con' as URI string instead.)r   )ÚcopyÚ
isinstancer    r   ÚconfigÚSQLALCHEMY_AVAILABLEÚsysÚmodulesr&   r   Ú
SelectableÚcreate_engineÚsplitÚcompiler'   Ú	TypeErrorÚtypeÚidÚloggerÚinfor   Úcreate_config_id)	r   r   r   r   r&   ÚengineÚsql_strr   r   s	   &&&     €r   r7   ÚSqlConfig.create_config_id-   s^  ø€ ð
 &×*Ñ*Ó,ˆð ˜EÕ"ˆÜ˜#œs×#Ò#Ü�‰×3×3Ð3¸ÌÏÉÔ8SÛ!ä˜c§>¡>×#<Ñ#<×=Ò=Ø'×5Ò5°mÀEÕ6J×6PÑ6PÐQVÓ6WÐXYÕ6ZÐ]bÕ6bÓc�FÜ! #§+¡+°f·n±n +Ó"EÓF�GØ+2 %Ò(ä#ØeÔfjÐknÓfoÐepÐprÐsvÐrwÐxóð ô  ØaÔbfÐgjÓbkÐalÐlnÐorÐnsÐtóð ð ˜EÕ"ˆÜ˜#œs×#Ò#Ü#% c£7ˆM˜%Ñ Ü�K‰KØ/´°S³	¨{ð  ;Pð  Qôô ‰wÑ'¨Ð'ÓWÐWr   c           
     ó‚   € R V P                   RV P                  RV P                  RV P                  RV P                  /pV# ©Ú	index_colÚcolumnsÚparamsÚcoerce_floatÚparse_datesr<   )r   Úpd_read_sql_kwargss   & r   rB   ÚSqlConfig.pd_read_sql_kwargsP   sE   € ð ˜Ÿ™Ø�t—|‘|Ø�d—k‘kØ˜D×-Ñ-Ø˜4×+Ñ+ð
Ðð "Ð!r   c                óŒ  <€ V ^8„  d   Qh/ S[ S[R3,          ;R&   S[ S[RRR3,          ;R&   S[S[ S[S[S[,          3,          ,          ;R&   S[;R&   S[S[ S[S[S[3,          ,          ;R	&   S[S[ S[S[3,          ,          ;R
&   S[S[S[,          ,          ;R&   S[S[,          ;R&   S[S[P                  ,          ;R&   # )r   zsqlalchemy.sql.Selectabler   zsqlalchemy.engine.Connectionzsqlalchemy.engine.Enginezsqlite3.Connectionr   r=   r@   r?   rA   r>   Ú	chunksizeÚfeatures)
r   r    r   ÚlistÚboolÚtupler   Úintr   r   )r!   r"   s   "€r   r#   r$      sß   ø‡ ‚ ñ 
‰sÐ/Ð/Õ	0Ñ7ñ	 ñ
 
‰sÐ2Ð4NÐPdÐdÕ	eÑlñ ñ ™™c¡4©¥9˜nÕ-Õ.Ñ5ñ ñ Ññ ñ ‘U™4¡©Ð,Õ-Õ.Ñ5ñ ñ ™%¡¡d 
Õ+Õ,Ñ3ñ ñ ‘d™3•iÕ Ñ'ñ ñ ™�}Ñ%ñ ñ ‘x×(Ñ(Õ)Ñ0ò r   © ©N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__r   r   r=   r@   r?   rA   r>   rE   rF   r   r7   ÚpropertyrB   Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__Ú__classcell__)r   r"   s   @@r   r   r      sc   ù‡ € á à37€CØhl€CØ15€IØ€LØ15€FØ/3€KØ#'€GØ%€IØ,0€Hõ6÷!Xõ !XðF ñ"ó ð"÷s … r   r   c                   óH   a € ] tR t^\t o ]tR tR tV 3R lR ltR t	Rt
V tR# )ÚSqlc                óX   € \         P                  ! V P                  P                  R 7      # ))rF   )r   ÚDatasetInfor*   rF   )r   s   &r   Ú_infoÚ	Sql._info_   s   € Ü×#Ò#¨T¯[©[×-AÑ-AÔBÐBr   c                ód   € \         P                  ! \         P                  P                  / R 7      .# ))ÚnameÚ
gen_kwargs)r   ÚSplitGeneratorÚSplitÚTRAIN)r   Ú
dl_managers   &&r   Ú_split_generatorsÚSql._split_generatorsb   s"   € Ü×'Ò'¬X¯^©^×-AÑ-AÈbÔQÐRÐRr   c                óN   <€ V ^8„  d   QhRS[ P                  RS[ P                  /# )r   Úpa_tabler   )ÚpaÚTable)r!   r"   s   "€r   r#   ÚSql.__annotate__e   s#   ø€ ÷ 	ñ 	¡B§H¡Hð 	±·±ñ 	r   c                ó  € V P                   P                  eí   V P                   P                  P                  p\        ;QJ d>    R V P                   P                  P	                  4        4       F  '       d   K   RM3	  RM/! R V P                   P                  P	                  4        4       4      '       dD   \
        P                  P                  V Uu. uF  q1VP                  ,          NK  	  upVR7      pV# \        W4      pV# u upi )Nc              3   óB   "  € T F  p\        V4      '       * x € K  	  R # 5irL   r   )Ú.0Úfeatures   & r   Ú	<genexpr>Ú"Sql._cast_table.<locals>.<genexpr>h   s   é € ÐbÑDa¸Ô+¨GÓ4×4Ñ4ÓDaùs   ‚FT)Úschema)
r*   rF   Úarrow_schemaÚallÚvaluesrh   ri   Úfrom_arraysr^   r
   )r   rg   rq   Úfields   &&  r   Ú_cast_tableÚSql._cast_tablee   sÂ   € Ø�;‰;×ÑÒ+Ø—[‘[×)Ñ)×6Ñ6ˆFß‹sÑbÀDÇKÁK×DXÑDX×D_ÑD_ÔDaÓb�s�sŠsÑbÀDÇKÁK×DXÑDX×D_ÑD_ÔDaÓb×bÒbäŸ8™8×/Ñ/ÑSYÓ0ZÑSYÈ%¸%¿*¹*×1EÐ1EÑSYÑ0ZÐciÐ/Ój�ð ˆô & hÓ7�Øˆùò	 1[s   ÃDc              #  óœ  "  € V P                   P                  p\        P                  ! V P                   P                  V P                   P
                  3R V/V P                   P                  B pVf   V.MTp\        V4       FC  w  r4\        P                  P                  V4      p\        ^ V4      V P                  V4      3x € KE  	  R# 5i)rE   N)r*   rE   ÚpdÚread_sqlr   r   rB   Ú	enumeraterh   ri   Úfrom_pandasr   rw   )r   rE   Ú
sql_readerÚ	chunk_idxÚdfrg   s   &     r   Ú_generate_tablesÚSql._generate_tablesp   s£   é € Ø—K‘K×)Ñ)ˆ	Ü—[’[Ø�K‰K�O‰O˜TŸ[™[Ÿ_™_ñ
Ø8Að
ØEIÇ[Á[×EcÑEcñ
ˆ
ð &/Ò%6�j‘\¸Jˆ
Ü& zÖ2‰MˆIÜ—x‘x×+Ñ+¨BÓ/ˆHÜ�a˜Ó# T×%5Ñ%5°hÓ%?Ð?Ô?ó 3ùs   ‚C
CrK   N)rM   rN   rO   rP   r   ÚBUILDER_CONFIG_CLASSr[   rd   rw   r�   rT   rU   )r"   s   @r   rX   rX   \   s-   ø‡ € Ø$ÐòCòS÷	ð 	÷@ð @r   rX   )r,   Údataclassesr   Útypingr   r   r   Úpandasrz   Úpyarrowrh   r   Údatasets.configÚdatasets.builderr   Údatasets.features.featuresr	   Údatasets.tabler
   Úsqlite3r&   ÚutilsÚloggingÚ
get_loggerrM   r5   ÚBuilderConfigr   ÚArrowBasedBuilderrX   rK   r   r   Ú<module>r’      s†   ðÛ 
Ý !ß 1Ñ 1ã Û ã Û Ý  Ý ;Ý %÷ Ûãð 
�‰×	Ñ	×	*Ñ	*¨8Ó	4€ð ô@"�×&Ñ&ó @"ó ð@"ôF@ˆ(×
$Ñ
$ö @r   