ó
    "EñiÚ  ã                   óT   • S r SSKJr   " S S5      r " S S\5      r " S S\5      rg	)
zÈContains definitions of the methods used by the _BaseDataLoaderIter to fetch data from an iterable-style or map-style dataset.

This logic is shared in both single- and multi-processing data loading.
é    )ÚNoReturnc                   ó,   • \ rS rSrSS jrS\4S jrSrg)Ú_BaseDatasetFetcheré
   ÚreturnNc                 ó4   • Xl         X l        X0l        X@l        g ©N)ÚdatasetÚauto_collationÚ
collate_fnÚ	drop_last)Úselfr
   r   r   r   s        ÚZ/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/utils/data/_utils/fetch.pyÚ__init__Ú_BaseDatasetFetcher.__init__   s   € ØŒØ,ÔØ$ŒØ"�ó    c                 ó   • [         er	   )ÚNotImplementedError)r   Úpossibly_batched_indexs     r   ÚfetchÚ_BaseDatasetFetcher.fetch   s   € Ü!Ð!r   )r   r   r
   r   ©r   N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r   r   Ú__static_attributes__© r   r   r   r   
   s   † ô#ð"¨x÷ "r   r   c                   ó2   ^ • \ rS rSrSU 4S jjrS rSrU =r$ )Ú_IterableDatasetFetcheré   c                 óT   >• [         TU ]  XX45        [        U5      U l        SU l        g )NF)Úsuperr   ÚiterÚdataset_iterÚended)r   r
   r   r   r   Ú	__class__s        €r   r   Ú _IterableDatasetFetcher.__init__   s%   ø€ Ü‰Ñ˜°*ÔHÜ  ›MˆÔØˆ�
r   c                 ó®  • U R                   (       a  [        eU R                  (       ao  / nU H(  n UR                  [	        U R
                  5      5        M*     [        U5      S:X  d)  U R                  (       a  [        U5      [        U5      :  a  [        eO[	        U R
                  5      nU R                  U5      $ ! [         a    SU l            Mz  f = f)NTr   )	r&   ÚStopIterationr   ÚappendÚnextr%   Úlenr   r   )r   r   ÚdataÚ_s       r   r   Ú_IterableDatasetFetcher.fetch   s©   € Ø�:�:ÜÐà××ØˆDÛ+�ðØ—K‘K¤ T×%6Ñ%6Ó 7Ö8ñ ,ô �4‹y˜A‹~Ø——¤3 t£9¬sÐ3IÓ/JÓ#Jä#Ð#øä˜×)Ñ)Ó*ˆDØ�‰˜tÓ$Ð$øô %ó Ø!%�D”JÛðús   ±$B>Â>CÃC)r%   r&   r   )r   r   r   r   r   r   r   Ú__classcell__)r'   s   @r   r    r       s   ø† ÷÷
%ð %r   r    c                   ó   • \ rS rSrS rSrg)Ú_MapDatasetFetcheré0   c                 óP  • U R                   (       aq  [        U R                  S5      (       a7  U R                  R                  (       a  U R                  R                  U5      nO.U Vs/ s H  o0R                  U   PM     nnOU R                  U   nU R	                  U5      $ s  snf )NÚ__getitems__)r   Úhasattrr
   r6   r   )r   r   r.   Úidxs       r   r   Ú_MapDatasetFetcher.fetch1   s€   € Ø××Ü�t—|‘| ^×4Ñ4¸¿¹×9R×9RØ—|‘|×0Ñ0Ð1GÓH‘á5KÓLÒ5K¨cŸ™ SÔ)Ñ5K�ÐL�à—<‘<Ð 6Ñ7ˆDØ�‰˜tÓ$Ð$ùò Ms   Á(B#r   N)r   r   r   r   r   r   r   r   r   r3   r3   0   s   † õ%r   r3   N)Ú__doc__Útypingr   r   r    r3   r   r   r   Ú<module>r<      s3   ðñõ
 ÷"ñ "ô%Ð1ô %ô6	%Ð,õ 	%r   