code: def test_dense_sample_with_warmup():
      run_dense_sample(
          iterations=100,
          expected=[[0.85635465, 1.8563547], [2.8563545, 3.8563545]],
          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 9 for calling a global function with example usage: test_dense_sample_with_warmup() without return types