avoid dict lookups

This commit is contained in:
J. Nick Koston
2023-02-15 23:37:28 -06:00
parent 9f54c30075
commit d3ec05f927

View File

@@ -10,7 +10,7 @@ from functools import lru_cache, partial
from itertools import chain, groupby from itertools import chain, groupby
import json import json
import logging import logging
from operator import attrgetter from operator import itemgetter
import os import os
import re import re
from statistics import mean from statistics import mean
@@ -2026,7 +2026,7 @@ def _sorted_statistics_to_dict(
start_ts_ = field_map["start_ts"] start_ts_ = field_map["start_ts"]
stats_by_meta_id: dict[int, list[Row]] = {} stats_by_meta_id: dict[int, list[Row]] = {}
seen_statistic_ids: set[str] = set() seen_statistic_ids: set[str] = set()
key_func = attrgetter("metadata_id") key_func = itemgetter(metadata_id_)
for meta_id, group in groupby(stats, key_func): for meta_id, group in groupby(stats, key_func):
stats_list = stats_by_meta_id[meta_id] = list(group) stats_list = stats_by_meta_id[meta_id] = list(group)
seen_statistic_ids.add(metadata[meta_id]["statistic_id"]) seen_statistic_ids.add(metadata[meta_id]["statistic_id"])