code: def test_schedulers():
      lr_scheduler = tf.keras.optimizers.schedules.ExponentialDecay(
          1e-3, 50, 0.5
      )
      wd_scheduler = tf.keras.optimizers.schedules.InverseTimeDecay(
          2e-3, 25, 0.25
      )
  
      run_dense_sample(
          iterations=100,
          expected=[[0.84216374, 1.8420818], [2.8420012, 3.841918]],
          optimizer=AdaBelief(
              learning_rate=lr_scheduler,
              weight_decay=wd_scheduler,
              rectify=False,
          ),
      )
  
imports: import numpy
  import pytest
  from packaging.version import Version
  import tensorflow.compat.v2
  from tensorflow_addons.optimizers import AdaBelief, Lookahead
label_desc: Use image node_id 17 for calling a global function with example usage: test_schedulers() without return types