Update requirements.txt (#1869)

* Update requirements.txt

* Add wandb.errors.UsageError

* bug fix
This commit is contained in:
Glenn Jocher
2021-12-01 15:37:56 +01:00
committed by GitHub
parent c35400cffd
commit 9d0e1cf298
2 changed files with 5 additions and 2 deletions
+4 -1
View File
@@ -24,7 +24,10 @@ try:
assert hasattr(wandb, '__version__') # verify package import not local dir
if pkg.parse_version(wandb.__version__) >= pkg.parse_version('0.12.2') and RANK in [0, -1]:
wandb_login_success = wandb.login(timeout=30)
try:
wandb_login_success = wandb.login(timeout=30)
except wandb.errors.UsageError: # known non-TTY terminal issue
wandb_login_success = False
if not wandb_login_success:
wandb = None
except (ImportError, AssertionError):