
    ȅi                     f    S r SSKJrJr  SSKJr  SSKJr  SSKJ	r	  SSK
Jr  \ " S S	5      5       rg
)z
Configuration classes for checkpointer construction.

This module provides configuration dataclasses that consolidate all
configuration options needed to construct checkpointers.
    )	dataclassfield   )BarrierConfig)CheckpointProcessConfig)CheckpointWriterConfig)CheckpointStagerConfigc                   r    \ rS rSr% Sr\" \S9r\\S'   \" \	S9r
\	\S'   \" \S9r\\S'   \" \S9r\\S'   Srg	)
CheckpointerConfig   a>  
Configuration class for checkpointer construction.

This class consolidates the core component configuration options needed to construct
a checkpointer, providing a clean separation of concerns where each component
manages its own configuration.

Attributes:
    writer_config: Configuration options for the checkpoint writer component.
    barrier_config: Configuration for barrier construction and arguments.
    staging_config: Configuration options for the async staging component.
    process_config: Configuration options for the async checkpoint process component.

)default_factorywriter_configbarrier_configstaging_configprocess_config N)__name__
__module____qualname____firstlineno____doc__r   r   r   __annotations__r   r   r	   r   r   r   __static_attributes__r       k/home/james-whalen/.local/lib/python3.13/site-packages/torch/distributed/checkpoint/_experimental/config.pyr   r      s]     -2.-M)  %*-$HNMH .3..N*  /4//N+ r   r   N)r   dataclassesr   r   barriersr   checkpoint_processr   checkpoint_writerr   stagingr	   r   r   r   r   <module>r!      s4    ) # 7 5 +   r   