| 83 | self.rank = rank |
| 84 | |
| 85 | def _func_wrapper(self, severity): |
| 86 | if severity == self.INTERNAL_ERROR: |
| 87 | return self._logger.critical |
| 88 | elif severity == self.ERROR: |
| 89 | return self._logger.error |
| 90 | elif severity == self.WARNING: |
| 91 | return self._logger.warning |
| 92 | elif severity == self.INFO: |
| 93 | return self._logger.info |
| 94 | elif severity == self.VERBOSE or severity == self.DEBUG: |
| 95 | return self._logger.debug |
| 96 | else: |
| 97 | raise AttributeError(f"No such severity: {severity}") |
| 98 | |
| 99 | @property |
| 100 | def trt_logger(self) -> trt.ILogger: |