From 986f77b75a7ea3fc81d11bcb8e9ac926c2b61dd0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 1 Dec 2021 15:15:27 +0100 Subject: [PATCH 1/2] `try: except (..., wandb.errors.UsageError)` --- utils/loggers/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index ae2d98bdc36d..280bcbf0c310 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -27,7 +27,7 @@ wandb_login_success = wandb.login(timeout=30) if not wandb_login_success: wandb = None -except (ImportError, AssertionError): +except (ImportError, AssertionError, wandb.errors.UsageError): wandb = None From bd71ee5de0ea83b985aeae2ee93ca3542193bc86 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 1 Dec 2021 15:26:54 +0100 Subject: [PATCH 2/2] bug fix --- utils/loggers/__init__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index 280bcbf0c310..2a68d9785071 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -24,10 +24,13 @@ 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, wandb.errors.UsageError): +except (ImportError, AssertionError): wandb = None