Check update_state signature for metrics.

tensorflow_addons/metrics/BUILD

@@ -90,3 +90,15 @@ py_test(
         ":metrics",
     ],
 )
+
+py_test(
+    name = "metrics_test",
+    size = "small",
+    srcs = [
+        "metrics_test.py",
+    ],
+    main = "metrics_test.py",
+    deps = [
+        ":metrics",
+    ],
+)

tensorflow_addons/metrics/metrics_test.py

@@ -0,0 +1,42 @@
+# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+import unittest
+import inspect
+
+from tensorflow.keras.metrics import Metric
+from tensorflow_addons import metrics
+
+
+class MetricsTests(unittest.TestCase):
+    def test_update_state_signature(self):
+        for name, obj in inspect.getmembers(metrics):
+            if inspect.isclass(obj) and issubclass(obj, Metric):
+                check_update_state_signature(obj)
+
+
+def check_update_state_signature(metric_class):
+    update_state_signature = inspect.signature(metric_class.update_state)
+    for expected_parameter in ["y_true", "y_pred", "sample_weight"]:
+        if expected_parameter not in update_state_signature.parameters.keys():
+            raise ValueError(
+                "Class {} is missing the parameter {} in the `update_state` "
+                "method. If the method doesn't use this argument, declare "
+                "it anyway and raise a UserWarning if it is "
+                "not None.".format(metric_class.__name__, expected_parameter)
+            )
+
+
+if __name__ == "__main__":
+    unittest.main()

tensorflow_addons/metrics/multilabel_confusion_matrix.py

@@ -14,6 +14,8 @@
 # ==============================================================================
 """Implements Multi-label confusion matrix scores."""
 
+import warnings
+
 import tensorflow as tf
 from tensorflow.keras.metrics import Metric
 import numpy as np
@@ -104,7 +106,14 @@ def __init__(
             dtype=self.dtype,
         )
 
-    def update_state(self, y_true, y_pred):
+    def update_state(self, y_true, y_pred, sample_weight=None):
+        if sample_weight is not None:
+            warnings.warn(
+                "`sample_weight` is not None. Be aware that MultiLabelConfusionMatrix "
+                "does not take `sample_weight` into account when computing the metric "
+                "value."
+            )
+
         y_true = tf.cast(y_true, tf.int32)
         y_pred = tf.cast(y_pred, tf.int32)
         # true positive

tensorflow_addons/metrics/r_square.py

@@ -14,6 +14,8 @@
 # ==============================================================================
 """Implements R^2 scores."""
 
+import warnings
+
 import tensorflow as tf
 from tensorflow.keras.metrics import Metric
 
@@ -53,7 +55,13 @@ def __init__(
         self.res = self.add_weight("residual", initializer="zeros")
         self.count = self.add_weight("count", initializer="zeros")
 
-    def update_state(self, y_true, y_pred):
+    def update_state(self, y_true, y_pred, sample_weight=None):
+        if sample_weight is not None:
+            warnings.warn(
+                "`sample_weight` is not None. Be aware that RSquare "
+                "does not take `sample_weight` into account when computing the metric "
+                "value."
+            )
         y_true = tf.convert_to_tensor(y_true, tf.float32)
         y_pred = tf.convert_to_tensor(y_pred, tf.float32)
         self.squared_sum.assign_add(tf.reduce_sum(y_true ** 2))