def _update_entity_metrics(self, loss: tf.Tensor, f1: tf.Tensor, tag_name: Text) -> None:
    if (tag_name == ENTITY_ATTRIBUTE_TYPE):
        self.entity_loss.update_state(loss)
        self.entity_f1.update_state(f1)
    elif (tag_name == ENTITY_ATTRIBUTE_GROUP):
        self.entity_group_loss.update_state(loss)
        self.entity_group_f1.update_state(f1)
    elif (tag_name == ENTITY_ATTRIBUTE_ROLE):
        self.entity_role_loss.update_state(loss)
        self.entity_role_f1.update_state(f1)
def _update_entity_metrics(self, loss: tf.Tensor, f1: tf.Tensor, tag_name: Text) -> None:
    if (tag_name == ENTITY_ATTRIBUTE_TYPE):
        self.entity_loss.update_state(loss)
        self.entity_f1.update_state(f1)
    elif (tag_name == ENTITY_ATTRIBUTE_GROUP):
        self.entity_group_loss.update_state(loss)
        self.entity_group_f1.update_state(f1)
    elif (tag_name == ENTITY_ATTRIBUTE_ROLE):
        self.entity_role_loss.update_state(loss)
        self.entity_role_f1.update_state(f1)