SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix WAND_DISABLED test (#9703)

Browse Source 
* Fix WAND_DISABLED test

* Remove duplicate import

* Make a test that actually works...

* Fix style
This commit is contained in:
Sylvain Gugger
2021-01-20 12:30:24 -05:00
committed by GitHub
parent 2a703773aa
commit 3cd91e8162
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/integrations.py
View File

@@ -54,7 +54,7 @@ from .trainer_utils import PREFIX_CHECKPOINT_DIR, BestRun, EvaluationStrategy #
# Integration functions: # Integration functions:
def is_wandb_available(): def is_wandb_available():
if os.getenv("WANDB_DISABLED"): if os.getenv("WANDB_DISABLED", "").upper() in ENV_VARS_TRUE_VALUES:
return False return False
return importlib.util.find_spec("wandb") is not None return importlib.util.find_spec("wandb") is not None