W&B: Enable login timeout (#4843)

* evolve fix

* Enable login timeout

* fix pkg
This commit is contained in:
Ayush Chaurasia authored and GitHub committed 2021-09-17 18:59:34 +02:00
1 parent 4fdaec0449
commit 302a1b0bb0
2 files changed
+4 -1

No files matched your search

+3
View File
@@ -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