+
    &j                        ^ RI t ^ RIHt ^ RIHtHt ^ RIHt ^ RIH	t	 ^ RI
t
^ RIHt . ROt] ! R R4      4       t ! R R]4      t] ! R R4      4       t] ! R R4      4       t] ! R R	4      4       t]],          t]]]]	,          3,          t] ! R R4      4       t] ! R R
4      4       t]! RR7       ! R R4      4       tR# )    N)Sequence)	dataclassfield)Enum)Any)	StatefulTChunkStorageMetadataTensorStorageMetadataBytesStorageMetadataMetadataMetadataIndexTensorPropertiesStorageMetac                   0   a  ] tR t^t o RtV 3R ltRtV tR# )r	   zc
Each chunk is expected to have the same properties of the TensorStorageMetadata
that includes it.
c                Z   < V ^8  d   Qh/ S[ P                  ;R&   S[ P                  ;R&   # )   offsetssizes)torchSize)format__classdict__s   "}/Users/jameslopez/projects/CWCArchive/cwc-podcast/.venv/lib/python3.14/site-packages/torch/distributed/checkpoint/metadata.py__annotate__!ChunkStorageMetadata.__annotate__   s'      ZZ  ::      N)__name__
__module____qualname____firstlineno____doc____annotate_func____static_attributes____classdictcell__r   s   @r   r	   r	      s       r   c                   &    ] tR t^"tRt^ t^t^tRtR# )_MEM_FORMAT_ENCODINGz'Describe the memory format of a tensor.r   N)	r   r   r    r!   r"   TORCH_CONTIGUOUS_FORMATTORCH_CHANNELS_LASTTORCH_PRESERVE_FORMATr$   r   r   r   r(   r(   "   s    1r   r(   c                      a  ] tR t^*t o Rt]! ]P                  R7      t]! ]P                  R7      t
Rt]! ]P                  R7      tRtR tR t]V 3R lR l4       tV 3R	 ltR
tV tR# )r   z)Properties used to create :class:`Tensor`default_factory)defaultFc                x   V P                   pV\        P                  8X  d   \        P                  pMZV\        P
                  8X  d   \        P                  pM4V\        P                  8X  d   \        P                  pM\        R V 24      hV P                  V P                  V P                  VV P                  3# )zInvalid torch.memory_format: )memory_formatr   contiguous_formatr(   r)   channels_lastr*   preserve_formatr+   RuntimeErrordtypelayoutrequires_grad
pin_memory)selfr1   mem_format_encodings   &  r   __getstate__TensorProperties.__getstate__9   s    **E333"6"N"Ne111"6"J"Je333"6"L"L!>}oNOO JJKKOO
 	
r   c                J   Vw  V n         V n        V n        pV n        V\        P
                  8X  d   \        P                  pMZV\        P                  8X  d   \        P                  pM4V\        P                  8X  d   \        P                  pM\        R V 24      hW0n        R# )z&Invalid torch.memory_format encoding: N)r6   r7   r8   r9   r(   r)   r   r2   r*   r3   r+   r4   r5   r1   )r:   stater;   r1   s   &&  r   __setstate__TensorProperties.__setstate__M   s     	
JKO "6"N"NN!33M $8$L$LL!//M $8$N$NN!11M89L8MN  +r   c                8   < V ^8  d   QhRS[ P                  RR/# )r   tensorreturnr   )r   Tensor)r   r   s   "r   r   TensorProperties.__annotate__g   s      
 
5<< 
4F 
r   c                    \        V P                  V P                  V P                  \        P
                  V P                  4       R 7      # ))r6   r7   r8   r1   r9   )r   r6   r7   r8   r   r2   	is_pinned)rC   s   &r   create_from_tensor#TensorProperties.create_from_tensorf   s;    ,,== ..11'')
 	
r   c                   < V ^8  d   Qh/ S[ P                  ;R&   S[ P                  ;R&   S[;R&   S[ P                  ;R&   S[;R&   # )r   r6   r7   r8   r1   r9   )r   r6   r7   boolr1   )r   r   s   "r   r   rF   *   sT     
 ;;G  LL7    &&O   r   )r6   r7   r1   r9   r8   N)r   r   r    r!   r"   r   r   get_default_dtyper6   stridedr7   r8   r2   r1   r9   r<   r@   staticmethodrI   r#   r$   r%   r&   s   @r   r   r   *   sh     3 u/F/FGE 7FM).u7N7N)OMJ
(+2 
 
{  r   c                   ,   a  ] tR t^qt o V 3R ltRtV tR# )r
   c                b   < V ^8  d   Qh/ S[ ;R&   S[P                  ;R&   S[S[,          ;R&   # )r   
propertiessizechunks)r   r   r   listr	   )r   r   s   "r   r   "TensorStorageMetadata.__annotate__q   s4         **  %&&	 r   r   N)r   r   r    r!   r#   r$   r%   r&   s   @r   r
   r
   q   s      r   c                       ] tR t^xtRtR# )r   r   N)r   r   r    r!   r$   r   r   r   r   r   x   s    r   c                   J   a  ] tR t^t o RtRtRt]! ]R7      t	V 3R lt
RtV tR# )r   Nr-   c                   < V ^8  d   Qh/ S[ S[P                  ,          R,          ;R&   S[ R,          ;R&   S[ R,          ;R&   S[S[ ,          ;R&   # )r   Ncheckpoint_idsave_idload_idmodules)strosPathLikerU   )r   r   s   "r   r   StorageMeta.__annotate__   sQ     $t+2  4Z  4Z	 
 #Y4 r   r   )r   r   r    r!   rZ   r[   r\   r   rU   r]   r#   r$   r%   r&   s   @r   r   r      s%     .2MGGt4G  r   c                   @   a  ] tR t^t o RtRtRtRtRtV 3R lt	Rt
V tR# )r   z5This class represents the metadata of the checkpoint.Nc                   < V ^8  d   Qh/ S[ S[S[3,          ;R&   S[;R&   S[;R&   S[R,          ;R&   S[R,          ;R&   # )r   state_dict_metadataplanner_datastorage_dataNstorage_metaversion)dictr^   STORAGE_TYPESr   r   )r   r   s   "r   r   Metadata.__annotate__   sY     
 c=011      $+  4Z r   r   )r   r   r    r!   r"   re   rf   rg   rh   r#   r$   r%   r&   s   @r   r   r      s%     ? LL'+LG  r   T)frozenc                   f   a  ] tR t^t o Rt Rt ]! RRRR7      t R	V 3R lR lltV 3R lt	Rt
V tR# )
r   zIThis class represents a lookup key for items in a state dict or Metadata.NF)hashcomparer/   c                X   < V ^8  d   QhRS[ RS[S[,          R,          RS[R,          /# )r   fqnoffsetNindex)r^   r   int)r   r   s   "r   r   MetadataIndex.__annotate__   s;     
C 
C
C $
C Tz	
Cr   c                    \         P                  V R V4       \         P                  V RV4       Ve.   \         P                  V R\        P                  ! V4      4       R# R# )rq   rs   Nrr   )object__setattr__r   r   )r:   rq   rr   rs   s   &&&&r   __init__MetadataIndex.__init__   sN     	4,4%0tXuzz&/AB r   c                n   < V ^8  d   Qh/ S[ ;R&   S[P                  R,          ;R&   S[R,          ;R&   # )r   rq   Nrr   rs   )r^   r   r   rt   )r   r   s   "r   r   ru      s8      
H	  JJ$  :F r   r   )NN)r   r   r    r!   r"   rr   r   rs   ry   r#   r$   r%   r&   s   @r   r   r      s;     S - $FM5%FE	
C 
C-  r   )r	   r
   r   r   r   r   r   )r_   collections.abcr   dataclassesr   r   enumr   typingr   r   %torch.distributed.checkpoint.statefulr   __all__r	   r(   r   r
   r   rj   ri   r^   STATE_DICT_TYPEr   r   r   r   r   r   <module>r      s   	 $ (    ;   4  C
 C
 C
L ' ' ' 	 	 	 &(<<sIO+, 5 5 5    $C C Cr   