W&B: Enable login timeout (#4843)
* evolve fix * Enable login timeout * fix pkg
This commit is contained in:
1 parent
4fdaec0449
commit
302a1b0bb0
2 files changed
+4
-1
No files matched your search
@@ -5,6 +5,7 @@ import os
|
||||
import sys
|
||||
from contextlib import contextmanager
|
||||
from pathlib import Path
|
||||
import pkg_resources as pkg
|
||||
|
||||
import yaml
|
||||
from tqdm import tqdm
|
||||
@@ -20,6 +21,8 @@ try:
|
||||
import wandb
|
||||
|
||||
assert hasattr(wandb, '__version__') # verify package import not local dir
|
||||
if pkg.parse_version(wandb.__version__) >= pkg.parse_version('0.12.2'):
|
||||
wandb.login(timeout=30)
|
||||
except (ImportError, AssertionError):
|
||||
wandb = None
|
||||
|
||||
|
||||
Reference in new issue
Block a user