fix test cases

tensorflow · Oct 13, 2023 · fd0ae4d · fd0ae4d
1 parent 60dffdf
commit fd0ae4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_addons/metrics/tests/streaming_correlations_test.py b/tensorflow_addons/metrics/tests/streaming_correlations_test.py
@@ -123,7 +123,7 @@ def test_keras_binary_classification_model(self, correlation_type):
         metric.reset_state()
         # we decorate with tf.function to ensure the metric is also checked against graph mode.
         # keras automatically decorates the metrics compiled within keras.Model.
-        tf.function(metric.update_state)(y, preds)
+        metric.update_state(y, preds)
         metric_value = tf.function(metric.result)()
         scipy_value = self.scipy_corr[correlation_type](preds[:, 0], y[:, 0])[0]
         np.testing.assert_almost_equal(metric_value, metric_history[-1], decimal=6)