Update requirements.txt (ultralytics#1869)

* Update requirements.txt * Add wandb.errors.UsageError * bug fix
kit-ar · Dec 1, 2021 · 9d0e1cf · 9d0e1cf
1 parent c35400c
commit 9d0e1cf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -14,7 +14,7 @@ tqdm>=4.41.0
 
 # Logging -------------------------------------
 tensorboard>=2.4.1
-# wandb
+wandb
 
 # Plotting ------------------------------------
 pandas>=1.1.4

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):