o
    …wÖiw  ã                   @   s&   d dl mZ G dd„ deƒZdgZdS )é    )ÚCallbackc                   @   sœ   e Zd ZdZd dd„Zd dd„Zd dd	„Zd d
d„Zd dd„Zd dd„Z	d dd„Z
d dd„Zd dd„Zd dd„Zd dd„Zd dd„Zd dd„Zd dd„ZdS )!ÚBaseCallbackzÔBase callback ABC for NeMo lifecycle hooks (extends PTL callback).

    Implementers may override any subset of the following methods. All are
    optional no-op defaults to keep implementations lightweight.
    ÚreturnNc                 O   ó   dS )z#Called when the application starts.N© ©ÚselfÚargsÚkwargsr   r   úY/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/nemo/lightning/base_callback.pyÚon_app_start   ó   zBaseCallback.on_app_startc                 O   r   )z!Called when the application ends.Nr   r   r   r   r   Ú
on_app_end   r   zBaseCallback.on_app_endc                 O   r   )z(Called when model initialization starts.Nr   r   r   r   r   Úon_model_init_start#   r   z BaseCallback.on_model_init_startc                 O   r   )z&Called when model initialization ends.Nr   r   r   r   r   Úon_model_init_end'   r   zBaseCallback.on_model_init_endc                 O   r   )z-Called when dataloader initialization starts.Nr   r   r   r   r   Úon_dataloader_init_start,   r   z%BaseCallback.on_dataloader_init_startc                 O   r   )z+Called when dataloader initialization ends.Nr   r   r   r   r   Úon_dataloader_init_end0   r   z#BaseCallback.on_dataloader_init_endc                 O   r   )z,Called when optimizer initialization starts.Nr   r   r   r   r   Úon_optimizer_init_start5   r   z$BaseCallback.on_optimizer_init_startc                 O   r   )z*Called when optimizer initialization ends.Nr   r   r   r   r   Úon_optimizer_init_end9   r   z"BaseCallback.on_optimizer_init_endc                 O   r   )z&Called when checkpoint loading starts.Nr   r   r   r   r   Úon_load_checkpoint_start>   r   z%BaseCallback.on_load_checkpoint_startc                 O   r   )z$Called when checkpoint loading ends.Nr   r   r   r   r   Úon_load_checkpoint_endB   r   z#BaseCallback.on_load_checkpoint_endc                 O   r   )z%Called when checkpoint saving starts.Nr   r   r   r   r   Úon_save_checkpoint_startF   r   z%BaseCallback.on_save_checkpoint_startc                 O   r   )z#Called when checkpoint saving ends.Nr   r   r   r   r   Úon_save_checkpoint_endJ   r   z#BaseCallback.on_save_checkpoint_endc                 O   r   )z'Called when checkpoint saving succeeds.Nr   r   r   r   r   Úon_save_checkpoint_successN   r   z'BaseCallback.on_save_checkpoint_successc                 O   r   )z<Update callback-specific configuration after initialization.Nr   r   r   r   r   Úupdate_configS   r   zBaseCallback.update_config)r   N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      s     












r   N)Úlightning.pytorch.callbacksr   ÚPTLCallbackr   Ú__all__r   r   r   r   Ú<module>   s   
F