[*.py] Rename "Arguments:" to "Args:"

tensorflow_model_optimization/python/core/clustering/keras/cluster.py

@@ -98,7 +98,7 @@ def cluster_weights(to_cluster,
   ])
   ```
 
-  Arguments:
+  Args:
       to_cluster: A single keras layer, list of keras layers, or a
         `tf.keras.Model` instance.
       number_of_clusters: the number of cluster centroids to form when
@@ -195,7 +195,7 @@ def _cluster_weights(to_cluster, number_of_clusters, cluster_centroids_init,
   ])
   ```
 
-  Arguments:
+  Args:
       to_cluster: A single keras layer, list of keras layers, or a
         `tf.keras.Model` instance.
       number_of_clusters: the number of cluster centroids to form when
@@ -265,7 +265,7 @@ def strip_clustering(model):
 
   Only sequential and functional models are supported for now.
 
-  Arguments:
+  Args:
       model: A `tf.keras.Model` instance with clustered layers.
 
   Returns:

tensorflow_model_optimization/python/core/clustering/keras/clustering_callbacks.py

@@ -25,7 +25,7 @@ class ClusteringSummaries(tf.keras.callbacks.TensorBoard):
     This class is derived from tf.keras.callbacks.TensorBoard and just adds
     functionality to write histograms with batch-wise frequency.
 
-    Arguments:
+    Args:
         log_dir: The path to the directory where the log files are saved
         cluster_update_freq: determines the frequency of updates of the
           clustering histograms. Same behaviour as parameter update_freq of the

tensorflow_model_optimization/python/core/clustering/keras/experimental/cluster.py

@@ -79,7 +79,7 @@ def cluster_weights(to_cluster, number_of_clusters, cluster_centroids_init,
   ])
   ```
 
-  Arguments:
+  Args:
       to_cluster: A single keras layer, list of keras layers, or a
         `tf.keras.Model` instance.
       number_of_clusters: the number of cluster centroids to form when

tensorflow_model_optimization/python/core/keras/compat.py

@@ -31,7 +31,7 @@ def assign(ref, value, name=None):
 def initialize_variables(testcase):
   """Handle global variable initialization in TF 1.X.
 
-  Arguments:
+  Args:
     testcase: instance of tf.test.TestCase
   """
   if hasattr(tf, 'global_variables_initializer') and not tf.executing_eagerly():

tensorflow_model_optimization/python/core/keras/utils.py

@@ -34,7 +34,7 @@ def smart_cond(pred, true_fn=None, false_fn=None, name=None):  # pylint: disable
   If `pred` is a bool or has a constant value, we return either `true_fn()`
   or `false_fn()`, otherwise we use `tf.cond` to dynamically route to both.
 
-  Arguments:
+  Args:
     pred: A scalar determining whether to return the result of `true_fn` or
       `false_fn`.
     true_fn: The callable to be performed if pred is true.

tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve/cluster_utils.py

@@ -42,7 +42,7 @@ def _type_model(model):
 def _get_clustered_weights(cluster_indices, cluster_centroids):
   """This function is for generating clustered weights using centroids and cluster indices.
 
-  Arguments:
+  Args:
     cluster_indices: a variable representing cluster indices
     cluster_centroids: a variable representing cluster centroids
   Returns:
@@ -64,7 +64,7 @@ def strip_clustering_cqat(to_strip):
   After the CQAT is done, these variables should be removed and the layer
   with the clustered weights should be restored.
 
-  Arguments:
+  Args:
       to_strip: A `tf.keras.Model` instance with clustered layers or a
       `tf.keras.layers.Layer` instance
 

tensorflow_model_optimization/python/core/sparsity/keras/prune.py

@@ -124,7 +124,7 @@ def prune_low_magnitude(to_prune,
   upon inspection, the weights of checkpoints are not sparse
   (https://github.com/tensorflow/model-optimization/issues/206).
 
-  Arguments:
+  Args:
       to_prune: A single keras layer, list of keras layers, or a
         `tf.keras.Model` instance.
       pruning_schedule: A `PruningSchedule` object that controls pruning rate
@@ -212,7 +212,7 @@ def strip_pruning(model):
 
   Only sequential and functional models are supported for now.
 
-  Arguments:
+  Args:
       model: A `tf.keras.Model` instance with pruned layers.
 
   Returns:

tensorflow_model_optimization/python/core/sparsity/keras/test_utils.py

@@ -100,7 +100,7 @@ def model_type_keys():
 def list_to_named_parameters(param_name, options):
   """Convert list of options for parameter to input to @parameterized.named_parameters.
 
-  Arguments:
+  Args:
     param_name: name of parameter
     options: list of options for parameter