add default_bucketize_value assert for vocab_list

alibaba · Jan 20, 2025 · 1529dbe · 1529dbe
1 parent f2e907f
commit 1529dbe
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tzrec/features/feature.py b/tzrec/features/feature.py
@@ -619,7 +619,10 @@ def vocab_list(self) -> List[str]:
                     # `default_value` and <OOV> vocab to vocab_list
                     assert self.config.default_bucketize_value < len(
                         self.config.vocab_list
-                    ), "default_bucketize_value should be less than len(vocab_list)"
+                    ), (
+                        "default_bucketize_value should be less than len(vocab_list) "
+                        f"in {self.__class__.__name__}[{self.name}]"
+                    )
                     self._vocab_list = list(self.config.vocab_list)
                 else:
                     self._vocab_list = [self.config.default_value, "<OOV>"] + list(