+
    ÒÜ³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# )ÚTorchFormatterc                ó<   <€ \         SV `  WR 7       W0n        ^ RIpR# ))ÚfeaturesÚtoken_per_repo_idN)ÚsuperÚ__init__Útorch_tensor_kwargsÚtorch)Úselfr   r   r   r   Ú	__class__s   &&&, €Úd/home/luca/screen-checking/.venv/lib/python3.14/site-packages/datasets/formatting/torch_formatter.pyr   ÚTorchFormatter.__init__!   s   ø€ Ü‰Ñ (ÐÔPØ#6Ô Ýó    c                óö   aa€ ^ RI o\        S\        4      '       d]   S'       dU   \        ;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      # 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>Ú.TorchFormatter._consolidate.<locals>.<genexpr>*   s]   øé € ð á�Aô ˜1˜eŸl™lÓ+×iÐi°·±¸6À!½9¿?¹?Ñ0J×iÐiÈqÏwÉwÐZ`ÐabÕZc×ZiÑZiÑOiÔiÛùs   ƒ!A6¥'A6Á)A6FT)r   r   ÚlistÚallÚstack)r   r   r   s   &f@r   Ú_consolidateÚTorchFormatter._consolidate&   sZ   ù€ Ûä�fœd×#Ò#¯ß‹sõ áó�s�sŠsõ áó÷ ò ð —{’{ 6Ó*Ð*Øˆr   c                ó~  € ^ RI p\        V\        \        \	        R4      34      '       d   V# \        V\
        P                  \
        P                  34      '       dF   \
        P                  ! VP                  \
        P                  4      '       d   VP                  4       # / p\        V\
        P                  \
        P                  34      '       d“   \
        P                  ! VP                  \
        P                  4      '       d^   RVP                  /pVP                  \
        P                  \
        P                  39   d    VP!                  \
        P                  4      pMs\        V\
        P                  \
        P                  34      '       dD   \
        P                  ! VP                  \
        P"                  4      '       d   RVP$                  /p\&        P(                  '       d‘   R\*        P,                  9   d|   ^ RIp\        WP0                  P0                  4      '       dS   \
        P2                  ! V4      pVP4                  ^8X  d   VRR\
        P6                  3,          pVP9                  R	4      p\&        P:                  '       d1   R\*        P,                  9   d    ^ RIHp \        W4      '       d   V#  \&        PB                  '       d3   R\*        P,                  9   d   ^ RI"H#pH$p \        WV34      '       d   V# VPJ                  ! V3/ / VCV PL                  CB #   \@         d     Lvi ; i)
r   Nr   ÚPIL:NNNÚtorchvision)ÚVideoReaderÚ
torchcodec)ÚAudioDecoderÚVideoDecoder)é   r   é   )'r   r   ÚstrÚbytesÚtypeÚnpÚ	characterÚndarrayÚ
issubdtyper   ÚtolistÚnumberÚintegerÚint64Úuint16Úuint32ÚastypeÚfloatingÚfloat32r   ÚPIL_AVAILABLEÚsysÚmodulesÚ	PIL.ImageÚImageÚasarrayÚndimÚnewaxisÚ	transposeÚTORCHVISION_AVAILABLEÚtorchvision.ior)   ÚImportErrorÚTORCHCODEC_AVAILABLEÚtorchcodec.decodersr+   r,   Útensorr   )r   Úvaluer   Údefault_dtyper'   r)   r+   r,   s   &&      r   Ú
_tensorizeÚTorchFormatter._tensorize1   s#  € Ûä�eœc¤5¬$¨t«*Ð5×6Ò6ØˆLÜ˜¤§¡¬b¯j©jÐ9×:Ò:¼r¿}º}ÈUÏ[É[ÔZ\×ZfÑZf×?gÒ?gØ—<‘<“>Ð!àˆä�eœbŸi™i¬¯©Ð4×5Ò5¼"¿-º-ÈÏÉÔUW×U_ÑU_×:`Ò:`Ø$ e§k¡kÐ2ˆMð �{‰{œrŸy™y¬"¯)©)Ð4Ô4ØŸ™¤R§X¡XÓ.�øä˜¤§	¡	¬2¯:©:Ð6×7Ò7¼B¿MºMÈ%Ï+É+ÔWY×WbÑWb×<cÒ<cØ$ e§m¡mÐ4ˆMä××Ð E¬S¯[©[Ô$8Ûä˜%§¡§¡×1Ò1ÜŸ
š
 5Ó)�Ø—:‘: ”?Ø! ! Q¬¯
©
Ð"2Õ3�EàŸ™¨	Ó2�Ü×'×'Ð'¨M¼S¿[¹[Ô,HðÝ6ä˜e×1Ò1Ø �Lð 2ô ×&×&Ð&¨<¼3¿;¹;Ô+FßFä˜%°Ð!=×>Ò>Ø�à�|Š|˜EÑSÐ%R¨Ð%R¸×9QÑ9QÐ%RÑSÐSøô ô Ùðús   Ê+L. Ì.L<Ì;L<c                ó  € ^ RI p\        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  q0P                  V4      NK  	  up4      # MK\        V\        \        34      '       d0   T P                  V Uu. uF  q0P                  V4      NK  	  up4      # V P                  V4      # u upi u upi )r   NÚ	__array__)r   Úhasattrr   r   rS   r2   r4   r   Úobjectr$   Úrecursive_tensorizer!   ÚtuplerP   )r   Údata_structr   Ú	substructs   &&  r   Ú_recursive_tensorizeÚ#TorchFormatter._recursive_tensorize_   sÖ   € Ûô �; ×,Ò,´ZÀÏ\É\×5ZÒ5ZØ%×/Ñ/Ó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   ÂDÃDc                ó    <€ V ^8„  d   QhRS[ /# )r-   rX   )Údict)ÚformatÚ__classdict__s   "€r   Ú__annotate__ÚTorchFormatter.__annotate__m   s   ø€ ÷ Rñ R©tñ Rr   c                ó2   € \        V P                  VR R7      # )F)Úmap_list)r   rZ   )r   rX   s   &&r   rV   Ú"TorchFormatter.recursive_tensorizem   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   r`   ra   p   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   rg   Úrows   && r   Ú
format_rowÚTorchFormatter.format_rowp   sB   € Ø×(Ñ(Ó*×6Ñ6°xÓ@ˆØ×*Ñ*×5Ñ5°cÓ:ˆØ×'Ñ'¨Ó,Ð,r   c                ó8   <€ V ^8„  d   QhRS[ P                  RR/# )r-   rg   rh   útorch.Tensor)rj   rk   )r^   r_   s   "€r   r`   ra   u   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   )rn   Úextract_columnrp   Údecode_columnÚcolumn_namesrV   r$   )r   rg   r   s   && r   Úformat_columnÚTorchFormatter.format_columnu   sb   € Ø×+Ñ+Ó-×<Ñ<¸XÓFˆØ×-Ñ-×;Ñ;¸F×DYÑDYÐZ[ÕD\Ó]ˆØ×)Ñ)¨&Ó1ˆØ×"Ñ" 6Ó*ˆØˆr   c                ó:   <€ V ^8„  d   QhRS[ P                  RS[/# rf   ri   )r^   r_   s   "€r   r`   ra   |   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# rm   )rn   Úextract_batchrp   Údecode_batchrV   r$   )r   rg   ÚbatchÚcolumn_names   &&  r   Úformat_batchÚTorchFormatter.format_batch|   sf   € Ø×*Ñ*Ó,×:Ñ:¸8ÓDˆØ×,Ñ,×9Ñ9¸%Ó@ˆØ×(Ñ(¨Ó/ˆÛ ˆKØ!%×!2Ñ!2°5Õ3EÓ!FˆEÓñ !àˆr   )r   )NN)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r$   rP   rZ   rV   rs   r{   rƒ   Ú__static_attributes__Ú__classdictcell__Ú__classcell__)r   r_   s   @@r   r	   r	       sJ   ù‡ € ÷ò
	ò,Tò\,÷Rð R÷-ð -÷
ð ÷÷ ð r   r	   rv   )r@   Úcollections.abcr   Útypingr   Únumpyr2   Úpyarrowrj   Ú r   Úutils.py_utilsr   Ú
formattingr   r   r	   © r   r   Ú<module>r”      sA   ðó  Ý #Ý  ã Û å Ý 'Ý '÷ Ûôb�_ W¨n¸gÐ%EÕFö br   