Unverified Commit 3cd91e81 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix WAND_DISABLED test (#9703)

* Fix WAND_DISABLED test

* Remove duplicate import

* Make a test that actually works...

* Fix style
parent 2a703773
...@@ -54,7 +54,7 @@ from .trainer_utils import PREFIX_CHECKPOINT_DIR, BestRun, EvaluationStrategy # ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment