code: def test_on_update_end(self):
      self.trainer.lr_scheduler_callback.on_update_end()
      self.assertAlmostEqual(
          self.trainer.optimizer.param_groups[0]["lr"], 1e-02
      )
  
      self.trainer.lr_scheduler_callback.on_update_end()
      self.assertAlmostEqual(
          self.trainer.optimizer.param_groups[0]["lr"], 1e-03
      )
  
imports: import argparse
  import os
  import unittest
  from copy import deepcopy
  import torch
  from mmf.common.registry import registry
  from mmf.models.base_model import BaseModel
  from mmf.trainers.callbacks.lr_scheduler import LRSchedulerCallback
  from mmf.utils.configuration import load_yaml
  from omegaconf import OmegaConf
label_desc: Use image node_id 3 for calling the TestLogisticsCallback obj's underlying member method code with example usage: obj.test_on_update_end() without return types