Блог

Я использую интерпретацию для проекта, и во время тестов я предоставляю несколько образцов (просто для ускорения тестов), что приводит к предупреждению пользователя

 /usr/local/lib/python3.8/site-packages/interpret/glassbox/ebm/utils.py:91: UserWarning: Too few samples per class, adapting test size to guarantee 1 sample per class.
  warnings.warn(
  

Я пытался подавить это предупреждение до сих пор, но безуспешно, я даже пытался использовать карт-бланш ignore::UserWarning в своих предупреждениях о фильтрах pytest следующим образом:

 [tool:pytest]
addopts = -s --maxfail=1 --failed-first --color=yes
filterwarnings = ignore::_pytest.warning_types.PytestUnknownMarkWarning
                 ignore::UserWarning
                 ;ignore::UserWarning:interpret
norecursedirs = docs .mypy_cache
python_files = test_*.py
  

Я углубился в файл, который выдал предупреждение, где фактические строки, которые выдают предупреждение, следующие:

 if n_test_samples < y_uniq:  # pragma: no cover
    warnings.warn(
        "Too few samples per class, adapting test size to guarantee 1 sample per class."
    )
    test_size = y_uniq
  

которое должно быть UserWarning , так как это значение по умолчанию для warnings.warn. Тогда возникает вопрос, как можно было бы подавить такое предупреждение, поскольку то, что я использовал ранее, не работает для этого случая (в конфигурации pytest)?

Вот максимально изолированный пример вышеупомянутой проблемы (с использованием mark опции для имитации нашей настройки):

 from interpret.glassbox import ExplainableBoostingClassifier
from numpy import array, ravel
from pytest import mark


@mark.filterwarnings("ignore::UserWarning")
def test():
    features = array([
        [0.4, 0.6, 0.8],
        [0.3, 0.7, 0.9],
        [0.1, 0.5, 1.1],
        [0.2, 0.8, 1.0],
        [0.2, 6.4, 1.2],
        [0.5, 8.6, 0.9],
        [0.5, 7.9, 0.7],
        [0.4, 7.5, 0.9],
    ])
    labels = ravel(array([
        ["class_1"],
        ["class_1"],
        ["class_1"],
        ["class_1"],
        ["class_2"],
        ["class_2"],
        ["class_2"],
        ["class_2"],
    ]))

    model = ExplainableBoostingClassifier()
    model.fit(X=features, y=labels)

# call funcy, should be done through pytest though
test()
  

Скорее всего, ваша проблема в том, что у вас есть pytest.ini или tox.ini , который считывается pytest. Если какой-либо из этих файлов существует в вашем path и содержит [pytest] раздел, это заменяет [tool:pytest] раздел setup.cfg . Таким образом, решение заключается либо в удалении pytest.ini / tox.ini , либо в перемещении filterwarnings строки в pytest.ini :

 [pytest]
filterwarnings = ignore::UserWarning:interpret
  

Как упоминалось в комментариях, в документации pytest указано:

Использование setup.cfg не рекомендуется, за исключением очень простых случаев использования. файлы .cfg используют другой анализатор, чем pytest.ini и tox.ini, что может вызвать проблемы с отслеживанием.

Таким образом, второй вариант, вероятно, является лучшим.