Skip to content

Commit 8baa726

Browse files
authored
refactor: simplify token2idx creation with dict comprehension (#168)
Signed-off-by: ktro2828 <kotaro.uetake@tier4.jp>
1 parent 758e5c8 commit 8baa726

1 file changed

Lines changed: 6 additions & 7 deletions

File tree

t4_devkit/tier4.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -238,13 +238,12 @@ def __make_reverse_index__(self, verbose: bool) -> None:
238238
if verbose:
239239
print("Reverse indexing...")
240240

241-
token2idx: dict[str, dict[str, int]] = {}
242-
for schema in SchemaName:
243-
token2idx[schema.value] = {}
244-
for idx, table in enumerate(self.get_table(schema.value)):
245-
table: SchemaTable
246-
token2idx[schema.value][table.token] = idx
247-
self._token2idx = token2idx
241+
self._token2idx: dict[str, dict[str, int]] = {
242+
schema.value: {
243+
table.token: idx for idx, table in enumerate(self.get_table(schema.value))
244+
}
245+
for schema in SchemaName
246+
}
248247

249248
self._label2id: dict[str, int] = {
250249
category.name: idx for idx, category in enumerate(self.category)

0 commit comments

Comments
 (0)