| from pytorch_lightning.loggers import WandbLogger | |
| def get_wandb_logger(loggers): | |
| """Gets the wandb logger if it is the list of loggers otherwise returns None.""" | |
| wandb_logger = None | |
| for logger in loggers: | |
| if isinstance(logger, WandbLogger): | |
| wandb_logger = logger | |
| return wandb_logger | |