Forward metric name to `update_metric`

[antoinebrl]

🚀 Feature Request

Motivation

get_metrics returns a mapping between strings and Metrics. However the name isn't available in the update_metric. Checking for types might not be sufficient to uniquely characteristic a metric.

Implementation

The proposed new signature for the update_metric method of the ComposerModel:

- def update_metric(self, batch: Any, outputs: Any, metric: Metric) -> None:
+ def update_metric(self, batch: Any, outputs: Any, metric: Metric, name: str) -> None: