snap-stanford / ogb

Benchmark datasets, data loaders, and evaluators for graph machine learning
https://ogb.stanford.edu
MIT License
1.89k stars 397 forks source link

Added `tqdm.auto` so to have a nice looking bar on jupyter notebooks #430

Closed LucaCappelletti94 closed 1 year ago

LucaCappelletti94 commented 1 year ago

As per the title, since I am using OGB in jupyter notebooks, I'd rather have a nice-looking loading bar.

Since you already using tqdm, it's just a drop-in.

Best, Luca

weihua916 commented 1 year ago

Thanks! I am not so familiar with auto option. How does this affect non jupyter notebook users?

LucaCappelletti94 commented 1 year ago

Hi @weihua916, simply it switches automatically depending on the context. On jupyter notebooks, it will show an HTML-based widget, while in CLIs it will show the loading bar you are used to.

weihua916 commented 1 year ago

Thanks for the clarification. LGTM!