😘

huggingface/datasets でプログレスバーの表示を止める

2021/12/14に公開

huggingface/datasets を使っていた時に、データセットを shuffle するたびにプログレスバーが表示されてつらかった。

datasets.utils.logging.set_verbosity() あたりをいじったものの、やはりプログレスバーは表示されたまま。

最終的には、以下の Issue を見つけて対応できた。

https://github.com/huggingface/datasets/issues/2651

from datasets.utils import disable_progress_bar
disable_progress_bar()

ところが、実際にこれを使ってみると、次のバージョンでは deprecated になりますよという warning が出てしまった。

そこで Issue にも書いたとおり、以下のように書いたほうが今はよい。

from datasets.utils import set_progress_bar_enabled
set_progress_bar_enabled(False)

Discussion