我正在尝试导入一个使用tqdm的函数。在某些情况下,我在笔记本中使用此函数,而在其他情况下,我在脚本中使用它。有以下选项:
# my_lib.py
from tqdm.notebook import tqdm # option 1
from tqdm import tqdm # option 2
# some function that uses tqdm
def my_fn():
for i in tqdm(range(10)):
pass
当我使用选项1时,当在脚本run.py
中调用函数时,它会打印出奇怪的结果。
# run.py
from my_lib import my_fn
if __name__ == '__main__':
my_fn()
当我使用选项2时,当在notebook test.ipynb
中调用函数时,它会打印出奇怪的结果。
# test.ipynb
from my_lib import my_fn
my_fn()
如何在这个函数中使用tqdm,并使它在两种情况下都能正常工作?
1条答案
按热度按时间6ss1mwsb1#
迟做总比不做好。
只需要使用下面的import即可,
这将自动在
tqdm.notebook
和tqdm.std
之间进行选择。来源:https://tqdm.github.io/docs/shortcuts/#tqdmautonotebook