a
    do                     @   s4   d Z ddlmZmZ ddlmZ G dd deZdS )z_
Lambda Callback
^^^^^^^^^^^^^^^

Create a simple callback on the fly using lambda functions.

    )CallableOptional)Callbackc                '   @   s   e Zd ZdZdee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee ee d%ddZdS )LambdaCallbacka  
    Create a simple callback on the fly using lambda functions.

    Args:
        **kwargs: hooks supported by :class:`~pytorch_lightning.callbacks.callback.Callback`

    Example::

        >>> from pytorch_lightning import Trainer
        >>> from pytorch_lightning.callbacks import LambdaCallback
        >>> trainer = Trainer(callbacks=[LambdaCallback(setup=lambda *args: print('setup'))])
    N)%setupteardownon_fit_start
on_fit_endon_sanity_check_starton_sanity_check_endon_train_batch_starton_train_batch_endon_train_epoch_starton_train_epoch_endon_validation_epoch_starton_validation_epoch_endon_test_epoch_starton_test_epoch_endon_validation_batch_starton_validation_batch_endon_test_batch_starton_test_batch_endon_train_starton_train_endon_validation_starton_validation_endon_test_starton_test_endon_exceptionon_save_checkpointon_load_checkpointon_before_backwardon_after_backwardon_before_optimizer_stepon_before_zero_gradon_predict_starton_predict_endon_predict_batch_starton_predict_batch_endon_predict_epoch_starton_predict_epoch_endc&           (      C   s6   t   D ]&\}&}'|&dkrq
|'d ur
t| |&|' q
d S )Nself)localsitemssetattr)(r+   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   kv r1   t/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/pytorch_lightning/callbacks/lambda_function.py__init__)   s
    (zLambdaCallback.__init__)%NNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNN)__name__
__module____qualname____doc__r   r   r3   r1   r1   r1   r2   r      s                                        r   N)r7   typingr   r   Z$pytorch_lightning.callbacks.callbackr   r   r1   r1   r1   r2   <module>   s   