From ce6add8ecc88671c7fd1d7ff40be16e29c6745fd Mon Sep 17 00:00:00 2001 From: Nathan Raw Date: Mon, 30 Aug 2021 08:45:57 -0600 Subject: [PATCH] :bug: fix small model card bugs (#13310) * :bug: fix small model card bugs * :lipstick: style --- src/transformers/modelcard.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/src/transformers/modelcard.py b/src/transformers/modelcard.py index 7cb6f711ae..76f21ee9cb 100644 --- a/src/transformers/modelcard.py +++ b/src/transformers/modelcard.py @@ -426,12 +426,15 @@ class TrainingSummary: result["dataset"]["args"] = dataset_arg_mapping[ds_tag] if len(metric_mapping) > 0: + result["metrics"] = [] for metric_tag, metric_name in metric_mapping.items(): - result["metric"] = { - "name": metric_name, - "type": metric_tag, - "value": self.eval_results[metric_name], - } + result["metrics"].append( + { + "name": metric_name, + "type": metric_tag, + "value": self.eval_results[metric_name], + } + ) model_index["results"].append(result) @@ -446,7 +449,7 @@ class TrainingSummary: metadata = _insert_values_as_list(metadata, "tags", self.tags) metadata = _insert_values_as_list(metadata, "datasets", self.dataset_tags) metadata = _insert_values_as_list(metadata, "metrics", list(metric_mapping.keys())) - metadata["model_index"] = self.create_model_index(metric_mapping) + metadata["model-index"] = self.create_model_index(metric_mapping) return metadata