From 20c2d08424f5dde90ecb601df2a0a92de98344ce Mon Sep 17 00:00:00 2001 From: ktro2828 Date: Mon, 18 Aug 2025 16:12:30 +0900 Subject: [PATCH] refactor: simplify token2idx creation with dict comprehension Signed-off-by: ktro2828 --- t4_devkit/tier4.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/t4_devkit/tier4.py b/t4_devkit/tier4.py index 0888adb..91cafd5 100644 --- a/t4_devkit/tier4.py +++ b/t4_devkit/tier4.py @@ -238,13 +238,12 @@ def __make_reverse_index__(self, verbose: bool) -> None: if verbose: print("Reverse indexing...") - token2idx: dict[str, dict[str, int]] = {} - for schema in SchemaName: - token2idx[schema.value] = {} - for idx, table in enumerate(self.get_table(schema.value)): - table: SchemaTable - token2idx[schema.value][table.token] = idx - self._token2idx = token2idx + self._token2idx: dict[str, dict[str, int]] = { + schema.value: { + table.token: idx for idx, table in enumerate(self.get_table(schema.value)) + } + for schema in SchemaName + } self._label2id: dict[str, int] = { category.name: idx for idx, category in enumerate(self.category)