+
    ÒÜ³jÆ  ã                   óœ   € ^ RI t ^ RIHt ^ RIHt ^ RIt^ RIt^RI	H
t
 ^RIHt ^RIHt ]'       d   ^ RIt ! R R]]R	]3,          4      tR# )
é    N)ÚMapping)ÚTYPE_CHECKING)Úconfig)Ú
map_nested)ÚTensorFormatterc                   ó�   a a€ ] tR t^ t oRV 3R lltR tR tR tV3R lR ltV3R lR lt	V3R	 lR
 lt
V3R lR ltRtVtV ;t# )ÚTFFormatterc                ó<   <€ \         SV `  WR 7       W0n        ^ RIpR# ))ÚfeaturesÚtoken_per_repo_idN)ÚsuperÚ__init__Útf_tensor_kwargsÚ
tensorflow)Úselfr   r   r   ÚtfÚ	__class__s   &&&, €Úa/home/luca/screen-checking/.venv/lib/python3.14/site-packages/datasets/formatting/tf_formatter.pyr   ÚTFFormatter.__init__!   s   ø€ Ü‰Ñ (ÐÔPØ 0ÔÝó    c                ó°  aa€ ^ RI o\        S\        4      '       dº   S'       d²   \        ;QJ d     VV3R lS 4       F  '       d   K   RM	  RM! VV3R lS 4       4      '       d   SP                  ! S4      # \        ;QJ d     VV3R lS 4       F  '       d   K   RM	  RM! VV3R lS 4       4      '       d   SP
                  P	                  S4      # S# )r   Nc              3   óð   <"  € T Fk  p\        VSP                  4      ;'       dI    VP                  S^ ,          P                  8H  ;'       d!    VP                  S^ ,          P                  8H  x € Km  	  R# 5i)r   N)Ú
isinstanceÚTensorÚshapeÚdtype©Ú.0ÚxÚcolumnr   s   & €€r   Ú	<genexpr>Ú+TFFormatter._consolidate.<locals>.<genexpr>*   s[   øé € ð ÙpvÐkl”
˜1˜bŸi™iÓ(×fÐf¨Q¯W©W¸¸q½	¿¹Ñ-G×fÐfÈAÏGÉGÐW]Ð^_ÕW`×WfÑWfÑLfÔfÓpvùs   ƒ!A6¥'A6Á)A6FTc              3   óæ   <"  € T Ff  p\        VSP                  SP                  34      ;'       d8    VP                  ^8H  ;'       d!    VP                  S^ ,          P                  8H  x € Kh  	  R# 5i)é   N)r   r   ÚRaggedTensorÚndimr   r   s   & €€r   r!   r"   .   s`   øé € ð á�Aô ˜1˜rŸy™y¨"¯/©/Ð:Ó;×jÐjÀÇÁÈ!Á×jÐjÐPQ×PWÑPWÐ[aÐbcÕ[d×[jÑ[jÑPjÔjÛùs   ƒ-A1±A1Á)A1)r   r   ÚlistÚallÚstackÚragged)r   r    r   s   &f@r   Ú_consolidateÚTFFormatter._consolidate&   sž   ù€ Ûä�fœd×#Ò#¯ß‹sõ Ùpvó�s�sŠsõ Ùpvó÷ ò ð —x’x Ó'Ð'ß“õ áó——’õ áó÷ ò ð
 —y‘y—‘ vÓ.Ð.àˆr   c                ó<  € ^ RI pVf   V# / p\        V\        P                  \        P                  34      '       dE   \        P
                  ! VP                  \        P                  4      '       d   RVP                  /pMs\        V\        P                  \        P                  34      '       dD   \        P
                  ! VP                  \        P                  4      '       d   RVP                  /p\        P                  '       dU   R\        P                  9   d@   ^ RIp\        WP                   P                   4      '       d   \        P"                  ! V4      p\        P$                  '       d1   R\        P                  9   d    ^ RIHp \        W4      '       d   V#  \        P,                  '       d3   R\        P                  9   d   ^ RIHpHp \        WV34      '       d   V# VP4                  ! V3/ / VCV P6                  CB #   \*         d     Lvi ; i)r   Nr   ÚPILÚtorchvision)ÚVideoReaderÚ
torchcodec)ÚAudioDecoderÚVideoDecoder)r   r   ÚnpÚnumberÚndarrayÚ
issubdtyper   ÚintegerÚint64ÚfloatingÚfloat32r   ÚPIL_AVAILABLEÚsysÚmodulesÚ	PIL.ImageÚImageÚasarrayÚTORCHVISION_AVAILABLEÚtorchvision.ior0   ÚImportErrorÚTORCHCODEC_AVAILABLEÚtorchcodec.decodersr2   r3   Úconvert_to_tensorr   )r   Úvaluer   Údefault_dtyper.   r0   r2   r3   s   &&      r   Ú
_tensorizeÚTFFormatter._tensorize7   su  € ÛàŠ=ØˆLàˆä�eœbŸi™i¬¯©Ð4×5Ò5¼"¿-º-ÈÏÉÔUW×U_ÑU_×:`Ò:`Ø$ b§h¡hÐ/‰MÜ˜¤§	¡	¬2¯:©:Ð6×7Ò7¼B¿MºMÈ%Ï+É+ÔWY×WbÑWb×<cÒ<cØ$ b§j¡jÐ1ˆMä××Ð E¬S¯[©[Ô$8Ûä˜%§¡§¡×1Ò1ÜŸ
š
 5Ó)�Ü×'×'Ð'¨M¼S¿[¹[Ô,HðÝ6ä˜e×1Ò1Ø �Lð 2ô ×&×&Ð&¨<¼3¿;¹;Ô+FßFä˜%°Ð!=×>Ò>Ø�à×#Ò# EÑXÐ-W°Ð-WÀ×AVÑAVÐ-WÑXÐXøô ô Ùðús   Æ
H ÈHÈHc                ó2  € ^ RI p\        P                  '       dw   R\        P                  9   db   ^ RIp\        WP                  4      '       dC   V P                  VP                  4       P                  4       P                  4       R,          4      # \        VR4      '       d,   \        WP                  4      '       g   VP                  4       p\        V\        P                  4      '       dF   VP                   \"        8X  d0   T P%                  V Uu. uF  q@P'                  V4      NK  	  up4      # MK\        V\(        \*        34      '       d0   T P%                  V Uu. uF  q@P'                  V4      NK  	  up4      # V P                  V4      # u upi u upi )r   NÚtorchÚ	__array__© )r   r   ÚTORCH_AVAILABLEr=   r>   rM   r   r   rJ   ÚdetachÚcpuÚnumpyÚhasattrrN   r4   r6   r   Úobjectr+   Úrecursive_tensorizer'   Útuple)r   Údata_structr   rM   Ú	substructs   &&   r   Ú_recursive_tensorizeÚ TFFormatter._recursive_tensorizeY   s.  € Ûô ×!×!Ð! g´·±Ô&<Ûä˜+§|¡|×4Ò4Ø—‘ {×'9Ñ'9Ó';×'?Ñ'?Ó'A×'GÑ'GÓ'IÈ"Õ'MÓNÐNÜ�; ×,Ò,´ZÀÏYÉY×5WÒ5WØ%×/Ñ/Ó1ˆKä�k¤2§:¡:×.Ò.Ø× Ñ ¤FÔ*Ø×(Ñ(Ñ_jÓ)kÑ_jÐR[×*BÑ*BÀ9Ö*MÑ_jÑ)kÓlÐlð +ä˜¤d¬E ]×3Ò3Ø×$Ñ$Ñ[fÓ%gÑ[fÈi×&>Ñ&>¸yÖ&IÑ[fÑ%gÓhÐhØ�‰˜{Ó+Ð+ùò *lùâ%gs   ÄFÅFc                ó    <€ V ^8„  d   QhRS[ /# )é   rX   )Údict)ÚformatÚ__classdict__s   "€r   Ú__annotate__ÚTFFormatter.__annotate__l   s   ø€ ÷ Rñ R©tñ Rr   c                ó2   € \        V P                  VR R7      # )F)Úmap_list)r   rZ   )r   rX   s   &&r   rV   ÚTFFormatter.recursive_tensorizel   s   € Ü˜$×3Ñ3°[È5ÔQÐQr   c                ó:   <€ V ^8„  d   QhRS[ P                  RS[/# ©r]   Úpa_tableÚreturn©ÚpaÚTabler   )r_   r`   s   "€r   ra   rb   o   s   ø€ ÷ -ñ -¡2§8¡8ð -±ñ -r   c                ó˜   € V P                  4       P                  V4      pV P                  P                  V4      pV P	                  V4      # ©N)Únumpy_arrow_extractorÚextract_rowÚpython_features_decoderÚ
decode_rowrV   )r   rh   Úrows   && r   Ú
format_rowÚTFFormatter.format_rowo   sB   € Ø×(Ñ(Ó*×6Ñ6°xÓ@ˆØ×*Ñ*×5Ñ5°cÓ:ˆØ×'Ñ'¨Ó,Ð,r   c                ó8   <€ V ^8„  d   QhRS[ P                  RR/# )r]   rh   ri   ú	tf.Tensor)rk   rl   )r_   r`   s   "€r   ra   rb   t   s   ø€ ÷ ñ ¡b§h¡hð °;ñ r   c                óà   € V P                  4       P                  V4      pV P                  P                  W!P                  ^ ,          4      pV P                  V4      pV P                  V4      pV# )r   )ro   Úextract_columnrq   Údecode_columnÚcolumn_namesrV   r+   )r   rh   r    s   && r   Úformat_columnÚTFFormatter.format_columnt   sb   € Ø×+Ñ+Ó-×<Ñ<¸XÓFˆØ×-Ñ-×;Ñ;¸F×DYÑDYÐZ[ÕD\Ó]ˆØ×)Ñ)¨&Ó1ˆØ×"Ñ" 6Ó*ˆØˆr   c                ó:   <€ V ^8„  d   QhRS[ P                  RS[/# rg   rj   )r_   r`   s   "€r   ra   rb   {   s   ø€ ÷ ñ ¡R§X¡Xð ±'ñ r   c                óà   € V P                  4       P                  V4      pV P                  P                  V4      pV P	                  V4      pV F  pV P                  W#,          4      W#&   K  	  V# rn   )ro   Úextract_batchrq   Údecode_batchrV   r+   )r   rh   ÚbatchÚcolumn_names   &&  r   Úformat_batchÚTFFormatter.format_batch{   sf   € Ø×*Ñ*Ó,×:Ñ:¸8ÓDˆØ×,Ñ,×9Ñ9¸%Ó@ˆØ×(Ñ(¨Ó/ˆÛ ˆKØ!%×!2Ñ!2°5Õ3EÓ!FˆEÓñ !àˆr   )r   )NN)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r+   rJ   rZ   rV   rt   r|   r„   Ú__static_attributes__Ú__classdictcell__Ú__classcell__)r   r`   s   @@r   r	   r	       sJ   ù‡ € ÷ ò
ò" YòD,÷&Rð R÷-ð -÷
ð ÷÷ ð r   r	   rw   )r=   Úcollections.abcr   Útypingr   rS   r4   Úpyarrowrk   Ú r   Úutils.py_utilsr   Ú
formattingr   r   r   r	   rO   r   r   Ú<module>r“      sA   ðó  Ý #Ý  ã Û å Ý 'Ý '÷ Ûôa�/ '¨;¸Ð"?Õ@ö ar   