Function build_rows

tools/static_bench/make_rd_table.py:270–298

(data: OrderedDict)

def build_rows(data: OrderedDict) -> Tuple[List[Dict], List[str]]:
    dataset_labels: List[str] = []
    rows: List[Dict] = []
    row_index: Dict[Tuple[str, Optional[str]], int] = {}

    for dataset_key, methods in data.items():
        dataset_label = normalize_dataset_label(dataset_key)
        dataset_labels.append(dataset_label)
        for method, variants in methods.items():
            variant_items = list(variants.items())
            has_multiple = len(variant_items) > 1
            for variant_key, metrics in variant_items:
                normalized_variant = normalize_variant_key(method, variant_key, has_multiple)
                row_id = (method, normalized_variant)
                if row_id not in row_index:
                    row_index[row_id] = len(rows)
                    rows.append(
                        {
                            "row_id": row_id,
                            "method": method,
                            "normalized_variant": normalized_variant,
                            "display_name": format_display_name(method, normalized_variant),
                            "datasets": {dataset_label: metrics},
                        }
                    )
                else:
                    rows[row_index[row_id]]["datasets"][dataset_label] = metrics

    return rows, dataset_labels


def build_table(rows: Sequence[Dict], dataset_labels: Sequence[str]) -> str: