使用NeuralProphet时如何从外部禁用模块内的tqdm进度条

• 问题本质：NeuralProphet在数据加载、训练初始化、验证集推理等多个内部流程硬编码调用了tqdm，没有对外暴露这些环节的进度条控制参数，直接在模型初始化、fit/predict阶段传disable=True、leave=False只能控制少部分主流程进度条，无法覆盖所有内部调用。

方案1：提前全局锁定tqdm禁用状态（推荐，无副作用） #

必须在导入NeuralProphet模块前执行tqdm初始化方法的覆写，从根源阻止所有tqdm实例输出进度条：

# 以下3行代码必须放在所有neuralprophet相关导入语句之前
from tqdm import tqdm
from functools import partialmethod
tqdm.__init__ = partialmethod(tqdm.__init__, disable=True)

# 之后再正常导入NeuralProphet编写业务逻辑
from neuralprophet import NeuralProphet
# 示例代码
model = NeuralProphet()
metrics = model.fit(df, freq="D")
forecast = model.predict(model.make_future_dataframe(df, periods=30))

该方案不需要修改NeuralProphet源码，通过猴子补丁修改tqdm类的默认初始化参数，优先级高于NeuralProphet内部传入的所有tqdm配置参数，能100%屏蔽所有tqdm生成的进度条。

方案2：上下文管理器屏蔽输出（适合不想全局禁用tqdm的场景） #

如果项目其他模块还需要用到tqdm进度条，可以用输出重定向的方式，仅屏蔽NeuralProphet运行阶段的控制台输出：

import os
import sys
from contextlib import contextmanager

@contextmanager
def suppress_console_output():
    devnull = open(os.devnull, "w")
    original_stdout = sys.stdout
    original_stderr = sys.stderr
    sys.stdout = devnull
    sys.stderr = devnull
    try:
        yield
    finally:
        sys.stdout = original_stdout
        sys.stderr = original_stderr
        devnull.close()

# 仅将NeuralProphet相关执行逻辑包裹在上下文管理器中即可
with suppress_console_output():
    model = NeuralProphet()
    metrics = model.fit(df, freq="D")
    future_df = model.make_future_dataframe(df, periods=90)
    forecast_res = model.predict(future_df)

注意：该方案会同时屏蔽上下文内所有控制台输出，如果需要查看模型训练的报错、日志信息，不要使用该方案。

内容的提问来源于stack exchange，提问作者Aditya