Tensorflow Keras custom metrics error on update_state()

Question

Context

Using TF 1.15 with tf.Estimator interface to train and evaluate models. Trying to write a custom TF metric, using tf.keras.metric.Metric for that.<