capreolus-ir / capreolus

A toolkit for end-to-end neural ad hoc retrieval
https://capreolus.ai
Apache License 2.0
95 stars 32 forks source link

Remove progress bar when running in a notebook #11

Open andrewyates opened 4 years ago

andrewyates commented 4 years ago

We normally use multiple tqdm progress bars at different positions, which is not rendered correctly in Colab. We should replace these with sparse logging messages when running in a notebook. Does it also make sense to do this when running in a normal shell?

kevinmartinjos commented 3 years ago

This could be because we are importing the console version of tqdm. This might fix the issue by automatically importing the correct (consolve vs notebook) tqdm:

from tqdm.autonotebook import tqdm