code: def scheduler_tester(
      scheduler, ref_scheduler, max_iter, scheduler_args=[], atol=1e-6
  ):
      # Create scheduler
      s = scheduler(*scheduler_args)
      ref_s = ref_scheduler(*scheduler_args)
  
      # Check learning rate
      lr = [s.get_learning_rate(iter) for iter in range(max_iter)]
      ref_lr = [
          ref_s.get_learning_rate(iter) for iter in range(max_iter)
      ]
      assert_allclose(lr, ref_lr, atol=atol)
  
label_desc: Use image node_id 1 for calling a global function with example usage: scheduler_tester(scheduler, ref_scheduler, max_iter, scheduler_args, atol) without return types