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

* `try: except (..., wandb.errors.UsageError)` * bug fix
fhkiel-mlaip · Dec 1, 2021 · 284a14d · 284a14d
1 parent 57fadd4
commit 284a14d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py
@@ -24,7 +24,10 @@
 
     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):