uninstalled wandb raises AttributeError

13aa1741 · Lysandre · 6e603cb7 · 13aa1741
Commit 13aa1741 authored Jun 09, 2020 by Lysandre
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/trainer.py src/transformers/trainer.py +1 -1

No files found.
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -69,7 +69,7 @@ try:
        wandb.termwarn("W&B installed but not logged in.  Run `wandb login` or set the WANDB_API_KEY env variable.")
    else:
        _has_wandb = False if os.getenv("WANDB_DISABLED") else True
-except ImportError:
+except (ImportError, AttributeError):
    _has_wandb = False