Pylint with pytorch: есть ли способ сообщить Pylint, что нужно искать модуль в другом месте?

Question

Я использую pytorch, а pylint не распознает несколько функций, например: torch.stack, однако, если я делаю import torch._C as torch, то, кажется, работает нормально.

Если я это сделаю выше, фактические модули, которые существуют внутри torch пакета, например torch.cuda или torch.nn, необходимо импортировать по отдельности, так как простое выполнение torch.cuda будет указывать на torch._C.cuda и, следовательно, не будет работать,

Есть ли способ заставить pylint смотреть на torch и torch._C, когда я делаю import torch или даже когда он видит факел?Я не думаю, что я бы использовал torch для ссылки на любую другую вещь в моем коде.

Answer 1

Решением на данный момент является добавление torch к generated-members:

pylint --generated-members="torch.*" ...

или в pylintrc в разделе [TYPECHECK]:

generated-members=torch.*

Я нашелэто решение в ответе на обсуждение github проблемы с pytorch [Незначительная ошибка] Pylint E1101 У модуля "torch" нет элемента "from_numpy" # 701 .Менее удовлетворяющий, чем белый список, потому что я думаю, что он не поймет, если вы ссылаетесь на то, что на самом деле не является членом, но это лучшее решение, с которым я когда-либо сталкивался.