code: def _write_stats(
      writers: List[MetricLogger],
      lr_stats: Dict[str, float],
      step: int,
  ) -> None:
      for writer in writers:
          writer.log_dict(lr_stats, step)
  
file_path: tnt/torchtnt/framework/callbacks/learning_rate_monitor.py
imports: from typing import Dict, List, Union
  from torchtnt.framework.callback import Callback
  from torchtnt.framework.state import State
  from torchtnt.framework.unit import TTrainUnit
  from torchtnt.utils.loggers.logger import MetricLogger
  from torchtnt.utils.optimizer import extract_lr_from_optimizer
label_desc: Use image node_id 1 for calling a global function with example usage: _write_stats(writers, lr_stats, step) without return types