o
    TÃiä  ã                   @   s,   d dl mZmZ d dlT G dd„ deƒZdS )é    )Úget_scalar_paramÚDeepSpeedConfigObject)Ú*c                       s$   e Zd Z‡ fdd„Zdd„ Z‡  ZS )ÚDeepSpeedNebulaConfigc                    sT   t t| ƒ ¡  d | _d | _d | _d | _d | _t| 	¡ v r!|t }ni }|  
|¡ d S ©N)Úsuperr   Ú__init__ÚenabledÚpersistent_storage_pathÚpersistent_time_intervalÚnum_of_version_in_retentionÚenable_nebula_loadÚNEBULAÚkeysÚ_initialize)ÚselfÚ
param_dictÚnebula_dict©Ú	__class__© úK/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/nebula/config.pyr      s   
zDeepSpeedNebulaConfig.__init__c                 C   sX   t |ttƒ| _t |ttƒ| _t |ttƒ| _	t |t
tƒ| _t |ttƒ| _t |ttƒ| _d S r   )r   ÚNEBULA_ENABLEDÚNEBULA_ENABLED_DEFAULTr	   ÚNEBULA_LOAD_PATHÚNEBULA_LOAD_PATH_DEFAULTÚ	load_pathÚNEBULA_ENABLE_NEBULA_LOADÚ!NEBULA_ENABLE_NEBULA_LOAD_DEFAULTr   ÚNEBULA_PERSISTENT_STORAGE_PATHÚ&NEBULA_PERSISTENT_STORAGE_PATH_DEFAULTr
   ÚNEBULA_PERSISTENT_TIME_INTERVALÚ'NEBULA_PERSISTENT_TIME_INTERVAL_DEFAULTr   Ú"NEBULA_NUM_OF_VERSION_IN_RETENTIONÚ*NEBULA_NUM_OF_VERSION_IN_RETENTION_DEFAULTr   )r   r   r   r   r   r      s   ÿÿÿ
ÿz!DeepSpeedNebulaConfig._initialize)Ú__name__Ú
__module__Ú__qualname__r   r   Ú__classcell__r   r   r   r   r   
   s    r   N)Údeepspeed.runtime.config_utilsr   r   Údeepspeed.nebula.constantsr   r   r   r   r   Ú<module>   s   