python - 设置 DEBUG 环境变量时，如何禁用 numba 调试日志记录？

Question

我的代码中有很多 numba 编译的函数和类。为了测试，我必须像这样运行我的程序:

DEBUG=True python3 main.py

但是当我像这样启动我的程序时，numba 会向 stderr 打印大量调试消息。这些消息如下所示:

2020-02-05 16:28:30,886 - numba.interpreter - DEBUG - label 0:
__numba_self_ = arg(0, name=__numba_self_) ['__numba_self_']
args = arg(1, name=args)                 ['args']
$4load_attr.1 = getattr(value=__numba_self_, attr=find) ['$4load_attr.1', '__numba_self_']
$8call_function_ex.3 = call $4load_attr.1(*args, func=$4load_attr.1, args=[], kws=[], vararg=args) ['$4load_attr.1', '$8call_function_ex.3', 'args']
$10return_value.4 = cast(value=$8call_function_ex.3) ['$10return_value.4', '$8call_function_ex.3']
return $10return_value.4                 ['$10return_value.4']

如何在不取消设置 DEBUG 环境变量的情况下禁用此输出？

Answer 1

import logging

numba_logger = logging.getLogger('numba')
numba_logger.setLevel(logging.WARNING)

会做的工作。只需在脚本的开头调用即可!