
    sh                     \    S SK Jr  S SK7  S SKJrJr  \(       a  S SKJr  Sr " S S5      rg)	    )TYPE_CHECKING)*)_WaitCounter_WaitCounterTracker)SummaryWriterztorch.monitor.Statc                   4    \ rS rSrSrS	S jrS\SS4S jrSrg)
TensorboardEventHandler   a-  
TensorboardEventHandler is an event handler that will write known events to
the provided SummaryWriter.

This currently only supports ``torch.monitor.Stat`` events which are logged
as scalars.

Example:
    >>> # xdoctest: +REQUIRES(env:TORCH_DOCTEST_MONITOR)
    >>> # xdoctest: +REQUIRES(module:tensorboard)
    >>> from torch.utils.tensorboard import SummaryWriter
    >>> from torch.monitor import TensorboardEventHandler, register_event_handler
    >>> writer = SummaryWriter("log_dir")
    >>> register_event_handler(TensorboardEventHandler(writer))
returnNc                     Xl         g)z-
Constructs the ``TensorboardEventHandler``.
N_writer)selfwriters     j/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/monitor/__init__.py__init__ TensorboardEventHandler.__init__   s	         eventc                     UR                   [        :X  aV  UR                  R                  5        H7  u  p#U R                  R                  X#UR                  R                  5       S9  M9     g g )N)walltime)name
STAT_EVENTdataitemsr   
add_scalar	timestamp)r   r   kvs       r   __call__ TensorboardEventHandler.__call__$   sP    ::#

((*''u7P7P7R'S + $r   r   )r   r   r   N)	__name__
__module____qualname____firstlineno____doc__r   Eventr    __static_attributes__ r   r   r	   r	      s"     Te T Tr   r	   N)	typingr   torch._C._monitorr   r   torch.utils.tensorboardr   r   r	   r)   r   r   <module>r-      s'       ? 5!
T Tr   