code: def test_keras_fit_with_schedule():
      """Check if calling model.fit works with wd schedule."""
      model = tf.keras.models.Sequential([tf.keras.layers.Dense(2)])
      loss = tf.keras.losses.SparseCategoricalCrossentropy(
          from_logits=True
      )
      wd_schedule = tf.keras.optimizers.schedules.ExponentialDecay(
          WEIGHT_DECAY, decay_steps=10, decay_rate=0.9
      )
      optimizer = weight_decay_optimizers.AdamW(
          learning_rate=1e-4, weight_decay=wd_schedule
      )
      model.compile(
          optimizer=optimizer, loss=loss, metrics=["accuracy"]
      )
      x, y = np.random.uniform(size=(2, 4, 1))
      model.fit(x, y, epochs=1)
  
imports: import importlib
  import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.utils import test_utils
  from tensorflow_addons.optimizers import weight_decay_optimizers
label_desc: Use image node_id 13 for calling a global function with example usage: test_keras_fit_with_schedule() without return types