From 1c19b423bf274a465f95725a79819bf82f71329e Mon Sep 17 00:00:00 2001 From: Boris Dayma Date: Fri, 8 Jan 2021 20:32:02 +0100 Subject: [PATCH] fix(wandb): fix config (#9489) --- src/transformers/integrations.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/integrations.py b/src/transformers/integrations.py index 0c541b846d..52ea9aa811 100644 --- a/src/transformers/integrations.py +++ b/src/transformers/integrations.py @@ -333,9 +333,9 @@ class WandbCallback(TrainerCallback): logger.warning( "W&B installed but not logged in. Run `wandb login` or set the WANDB_API_KEY env variable." ) - self._wandb = wandb - else: self._wandb = None + else: + self._wandb = wandb self._initialized = False def setup(self, args, state, model, reinit, **kwargs):