+
    QV-jO  ã                   ó°   € ^ RI t ^ RIHtHtHt ^RIHtHtHt ^RI	H
t
 ^RIHt ^RIHt ^RIHt ]'       d	   ^ RIt^ RIt ! R R	]4      t ! R
 R4      tR# )é    N)ÚTYPE_CHECKINGÚOptionalÚUnion)ÚDatasetÚFeaturesÚconfig)Úquery_table)ÚSql)Útqdm)ÚAbstractDatasetInputStreamc                   óH   a a€ ] tR t^t oRV3R lV 3R llltR tRtVtV ;t# )ÚSqlDatasetReaderc          
      ót   <€ V ^8„  d   QhRS[ S[R3,          RS[ S[RRR3,          RS[S[,          RS[R	S[/# )
é   Úsqlzsqlalchemy.sql.SelectableÚconúsqlalchemy.engine.Connectionúsqlalchemy.engine.Engineúsqlite3.ConnectionÚfeaturesÚ	cache_dirÚkeep_in_memory)r   Ústrr   r   Úbool)ÚformatÚ__classdict__s   "€Ú`/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/datasets/io/sql.pyÚ__annotate__ÚSqlDatasetReader.__annotate__   s^   ø€ ÷ 
ñ 
á‘3Ð3Ð3Õ4ð
ñ ‘3Ð6Ð8RÐThÐhÕið
ñ ™8Õ$ð	
ñ
 ð
ñ ñ
ó    c                ód   <€ \         SV `  ! RR VRVRV/VB  \        RRVR VRVRV/VB V n        R# )r   r   r   r   r   N© )ÚsuperÚ__init__r
   Úbuilder)Úselfr   r   r   r   r   ÚkwargsÚ	__class__s   &&&&&&,€r   r$   ÚSqlDatasetReader.__init__   s`   ø€ ô 	‰ÒÑi (Ði°iÐiÐP^ÐiÐbhÒiÜñ 
Øð
àð
ð ð
ð ð	
ð
 ñ
ˆŽr    c                ó¢   € R pR pR pR pV P                   P                  VVVVR7       V P                   P                  RV P                  R7      pV# )N)Údownload_configÚdownload_modeÚverification_modeÚ	base_pathÚtrain)ÚsplitÚ	in_memory)r%   Údownload_and_prepareÚ
as_datasetr   )r&   r+   r,   r-   r.   Údatasets   &     r   ÚreadÚSqlDatasetReader.read$   s`   € ØˆØˆØ ÐØˆ	à�‰×)Ñ)Ø+Ø'Ø/Øð	 	*ô 	
ð —,‘,×)Ñ)°À4×CVÑCVÐ)ÓWˆØˆr    )r%   )NNF)	Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r$   r5   Ú__static_attributes__Ú__classdictcell__Ú__classcell__)r(   r   s   @@r   r   r      s   ù‡ € ÷
õ 
÷$ò r    r   c                   ó`   a € ] tR t^6t o R
V 3R lR lltV 3R lR ltR tV 3R lR ltR	tV t	R# )ÚSqlDatasetWriterNc                óp   <€ V ^8„  d   QhRS[ RS[RS[S[RRR3,          RS[S[,          RS[S[,          /# )	r   r4   Únamer   r   r   r   Ú
batch_sizeÚnum_proc)r   r   r   r   Úint)r   r   s   "€r   r   ÚSqlDatasetWriter.__annotate__7   sV   ø€ ÷ +ñ +áð+ñ ð+ñ ‘3Ð6Ð8RÐThÐhÕið	+ñ
 ™S•Mð+ñ ™3•-ñ+r    c                ó´   € Ve   V^ 8:  d   \        RV R24      hWn        W n        W0n        V'       d   TM\        P
                  V n        WPn        W`n        R # )Nz	num_proc z must be an integer > 0.)	Ú
ValueErrorr4   rA   r   r   ÚDEFAULT_MAX_BATCH_SIZErB   rC   Úto_sql_kwargs)r&   r4   rA   r   rB   rC   rI   s   &&&&&&,r   r$   ÚSqlDatasetWriter.__init__7   sP   € ð Ò H°¤MÜ˜y¨¨
Ð2JÐKÓLÐLàŒØŒ	ØŒß(2™*¼×8UÑ8UˆŒØ ŒØ*Ör    c                ó    <€ V ^8„  d   QhRS[ /# ©r   Úreturn©rD   )r   r   s   "€r   r   rE   J   s   ø€ ÷ ñ ‘sñ r    c                óê   € V P                   P                  R R4      pV P                   P                  RR4      pV P                   P                  RR4      pV P                  ! RRV/V P                   B pV# )r   Nr   ÚindexFr"   )rI   ÚpopÚ_write)r&   Ú_rP   Úwrittens   &   r   ÚwriteÚSqlDatasetWriter.writeJ   sk   € Ø×Ñ×"Ñ" 5¨$Ó/ˆØ×Ñ×"Ñ" 5¨$Ó/ˆØ×"Ñ"×&Ñ& w°Ó6ˆà—+’+Ñ@ EÐ@¨T×-?Ñ-?Ñ@ˆØˆr    c                ól  € Vw  r#pV^ 8”  d	   / VCRR/CMTp\        V P                  P                  \        W"V P                  ,           4      V P                  P
                  R7      pVP                  4       pVP                  ! V P                  V P                  3RV/VB pT;'       g    \        V4      # )r   Ú	if_existsÚappend)ÚtableÚkeyÚindicesrP   )r	   r4   ÚdataÚslicerB   Ú_indicesÚ	to_pandasÚto_sqlrA   r   Úlen)r&   ÚargsÚoffsetrP   rI   ÚbatchÚdfÚnum_rowss   &&      r   Ú
_batch_sqlÚSqlDatasetWriter._batch_sqlR   sž   € Ø'+Ñ$ˆ�}ØDJÈQÄJÐ@˜=Ð@¨+°xÒ@ÐTaˆÜØ—,‘,×#Ñ#Ü�f t§¡Õ6Ó7Ø—L‘L×)Ñ)ô
ˆð
 �_‰_ÓˆØ—9’9˜TŸY™Y¨¯©ÑO¸ÐOÀÑOˆØ×"Ð"œ3˜r›7Ð"r    c                ó    <€ V ^8„  d   QhRS[ /# rL   rN   )r   r   s   "€r   r   rE   ^   s   ø€ ÷ ñ ±ñ r    c                ó¸  € ^ pV P                   e   V P                   ^8X  dY   \        \        ^ \        V P                  4      V P
                  4      RRR7       F  pW0P                  WAV34      ,          pK  	  V# \        V P                  4      V P
                  re\        P                  ! V P                   4      ;_uu_ 4       p\        TP                  V P                  \        ^ WV4       Uu. uF  qDW3NK  	  up4      WV,          '       d   WV,          ^,           MWV,          RRR7       F  pW5,          pK  	  RRR4       V# u upi   + '       g   i     T# ; i)zrWrites the pyarrow table as SQL to a database.

Caller is responsible for opening and closing the SQL connection.
NÚbazCreating SQL from Arrow format)ÚunitÚdesc)Útotalrm   rn   )
rC   Úhf_tqdmÚrangerb   r4   rB   rh   ÚmultiprocessingÚPoolÚimap)r&   rP   rI   rT   rd   rg   rB   Úpools   &&,     r   rR   ÚSqlDatasetWriter._write^   s&  € ð
 ˆà�=‰=Ò  D§M¡M°QÔ$6Ü!Ü�aœ˜TŸ\™\Ó*¨D¯O©OÓ<ØØ5÷�ð
 Ÿ?™?¨F¸=Ð+IÓJÕJ’ñð( ˆô $' t§|¡|Ó#4°d·o±o�jÜ ×%Ò% d§m¡m×4Ô4¸Ü 'Ø—I‘IØŸ™ÜFKÈAÈxÔFdÓeÑFd¸F %Ó7ÑFdÑeóð ;C×:OÔ:O˜8Õ1°QÖ6ÐU]ÕUkØØ9÷!�Hð Õ'’Gñ!÷ 5ð ˆùò f÷	 5Ö4ð ˆús   Ã*EÃ,EÃ8AEÅEÅE	)rB   r   r4   rA   rC   rI   )NN)
r7   r8   r9   r:   r$   rU   rh   rR   r;   r<   )r   s   @r   r?   r?   6   s(   ø‡ € ÷+ò +÷&ð ò
#÷ö r    r?   )rr   Útypingr   r   r   Ú r   r   r   Ú
formattingr	   Úpackaged_modules.sql.sqlr
   Úutilsr   rp   Úabcr   Úsqlite3Ú
sqlalchemyr   r?   r"   r    r   Ú<module>r      sE   ðÛ ß 1Ñ 1ç (Ñ (Ý $Ý *Ý #Ý +÷ Ûãô"Ð1ô "÷JDó Dr    