Close logger and cleanup.
(self)
| 328 | self._log_to_wandb(wandb_stats) |
| 329 | |
| 330 | def close(self) -> None: |
| 331 | """Close logger and cleanup.""" |
| 332 | if self._wandb_enabled and HAS_WANDB: |
| 333 | try: |
| 334 | wandb.finish() |
| 335 | except Exception: |
| 336 | pass |
| 337 | |
| 338 | for handler in self._logger.handlers[:]: |
| 339 | handler.close() |
| 340 | self._logger.removeHandler(handler) |
| 341 | |
| 342 | |
| 343 | def get_logger(name: str = "datastudio") -> Logger: |
no outgoing calls