Fix metric function to deliver information to api doc generator.

PiperOrigin-RevId: 419512239

Author: rino20

tensorflow_model_optimization/python/core/keras/metrics.py

@@ -14,6 +14,8 @@
 # ==============================================================================
 """Implements monitoring."""
 
+import functools
+
 from tensorflow.python.eager import monitoring
 
 
@@ -61,6 +63,7 @@ def get_usage_gauge(self, name):
     raise ValueError('Invalid gauge name: {}'.format(name))
 
   def __call__(self, func):
+    @functools.wraps(func)
     def inner(*args, **kwargs):
       try:
         results = func(*args, **kwargs)

tensorflow_model_optimization/python/core/quantization/keras/quantize.py

@@ -295,7 +295,8 @@ def quantize_apply(
   quantization have already been annotated. See `quantize_annotate_model`
   and `quantize_annotate_layer`.
 
-  Quantize model.
+  Example:
+
   ```python
   model = keras.Sequential([
       layers.Dense(10, activation='relu', input_shape=(100,)),