code: def testEnd(self):
      step = 10
      lr = 0.05
      end_lr = 0.001
      decayed_lr = schedules.PolynomialDecay(lr, 10, end_lr)
      expected = end_lr
      self.assertAllClose(decayed_lr(step), expected, 1e-6)
  
file_path: model-optimization/tensorflow_model_optimization/python/core/common/keras/compression/schedules_test.py
imports: import tensorflow
  from tensorflow_model_optimization.python.core.common.keras.compression import schedules
label_desc: Use image node_id 2 for calling the LinearPolynomialDecayTest obj's underlying member method code with example usage: obj.testEnd() without return types