code: def test_serialization():
      optimizer = lamb.LAMB(
          1e-4,
          weight_decay_rate=0.01,
          exclude_from_weight_decay=["var1"],
      )
      config = tf.keras.optimizers.serialize(optimizer)
      new_optimizer = tf.keras.optimizers.deserialize(config)
      assert new_optimizer.get_config() == optimizer.get_config()
  
imports: import numpy
  from numpy import linalg
  import pytest
  import tensorflow
  from tensorflow_addons.optimizers import lamb
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 14 for calling a global function with example usage: test_serialization() without return types