Skip to content

Commit d3b2e4f

Browse files
author
Sharad
committed
fix(multi_datasets-evaluator): return single dict
1 parent b68dba4 commit d3b2e4f

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

mmocr/evaluation/evaluator/multi_datasets_evaluator.py

+7-4
Original file line numberDiff line numberDiff line change
@@ -92,15 +92,18 @@ def evaluate(self, size: int) -> dict:
9292
metrics_results.update(metric_results)
9393
metric.results.clear()
9494
if is_main_process():
95-
averaged_results = self.average_results(metrics_results)
95+
averaged_results = [self.average_results(metrics_results)]
9696
else:
97-
averaged_results = None
97+
averaged_results = [None]
9898

9999
metrics_results = [metrics_results]
100100
broadcast_object_list(metrics_results)
101101
broadcast_object_list([averaged_results])
102-
103-
return metrics_results[0], averaged_results
102+
results = {
103+
'metric_results': metrics_results[0],
104+
'averaged_results': averaged_results
105+
}
106+
return results
104107

105108
def average_results(self, metrics_results):
106109
"""Compute the average of metric results across all datasets.

0 commit comments

Comments
 (0)