apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update requirements.txt (#1869)

Browse Source 
* 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -14,7 +14,7 @@ tqdm>=4.41.0
# Logging -------------------------------------
tensorboard>=2.4.1
# wandb
wandb
# Plotting ------------------------------------
pandas>=1.1.4

5
utils/loggers/__init__.py
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):