Handle non-TTY wandb.errors.UsageError (#5839)

* `try: except (..., wandb.errors.UsageError)`

* bug fix
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-12-01 15:38:02 +01:00
1 parent a4207a202d
commit bc484579d7
1 file changed
+4 -1
+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):