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)