Commit ebc2b2b9 authored by Thibault Prouteau's avatar Thibault Prouteau
Browse files

Fixed problem with `tqdm_notebook`:

* Proper import of `tqdm.notebook.tqdm_notebook`
* Fixed `tqdm_version in requirements

Fixed problem with `None` object for `tqdm` progressbar in notebook.
parent 2fd954f3
...@@ -4,4 +4,4 @@ numpy == 1.16.4 ...@@ -4,4 +4,4 @@ numpy == 1.16.4
pandas == 0.24.2 pandas == 0.24.2
scipy == 1.0.0 scipy == 1.0.0
setuptools == 20.7.0 setuptools == 20.7.0
tqdm == 4.19.6 tqdm == 4.45.0
...@@ -13,7 +13,7 @@ from scipy.sparse.linalg import svds ...@@ -13,7 +13,7 @@ from scipy.sparse.linalg import svds
from scipy.stats import pearsonr from scipy.stats import pearsonr
from joblib import Parallel, delayed from joblib import Parallel, delayed
from collections import OrderedDict, Counter from collections import OrderedDict, Counter
from tqdm import tqdm, tqdm_notebook from tqdm import tqdm, notebook
from .utils import Utils from .utils import Utils
from .window import WindowWeights from .window import WindowWeights
...@@ -210,8 +210,8 @@ class svd2vec: ...@@ -210,8 +210,8 @@ class svd2vec:
# solves a bug in jupyter notebooks # solves a bug in jupyter notebooks
# https://github.com/tqdm/tqdm/issues/485#issuecomment-473338308 # https://github.com/tqdm/tqdm/issues/485#issuecomment-473338308
print('\r', end='', flush=True) print('\r', end='', flush=True)
func = tqdm_notebook if notebook else tqdm func = notebook.tqdm_notebook if notebook else tqdm
format = None if notebook else "{desc: <30} {percentage:3.0f}% {bar}" format = "{desc: <30} {percentage:3.0f}% {bar}"
return func( return func(
iterable=yielder, iterable=yielder,
desc=desc, desc=desc,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment