code: def test_sparse_sample_with_warmup():
      # Expected values are obtained from the previous implementation
      run_sparse_sample(
          iterations=200,
          expected=[[0.8502214, 2.0], [3.0, 3.85022]],
          optimizer=AdaBelief(
              lr=1e-3,
              total_steps=100,
              warmup_proportion=0.1,
              min_lr=1e-5,
              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 10 for calling a global function with example usage: test_sparse_sample_with_warmup() without return types