Fix WAND_DISABLED test (#9703)

* Fix WAND_DISABLED test * Remove duplicate import * Make a test that actually works... * Fix style

Fix WAND_DISABLED test (#9703)
* Fix WAND_DISABLED test * Remove duplicate import * Make a test that actually works... * Fix style
3cd91e81 · Sylvain Gugger · GitHub · 2a703773 · 3cd91e81
Unverified Commit 3cd91e81 authored Jan 20, 2021 by Sylvain Gugger Committed by GitHub Jan 20, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/src/transformers/integrations.py
+++ b/src/transformers/integrations.py
@@ -54,7 +54,7 @@ from .trainer_utils import PREFIX_CHECKPOINT_DIR, BestRun, EvaluationStrategy  #

 # Integration functions:
 def is_wandb_available():
-    if os.getenv("WANDB_DISABLED"):
+    if os.getenv("WANDB_DISABLED", "").upper() in ENV_VARS_TRUE_VALUES:
        return False
    return importlib.util.find_spec("wandb") is not None