pytorch的Pylint：有没有办法告诉Pylint在不同的地方寻找模块？

Question

我正在使用pytorch，而pylint不能识别ex的几个功能：torch.stack但是，如果我这样做了，import torch._C as torch似乎可以正常工作。

如果执行上述操作，则需要分别导入torch软件包（如torch.cuda或torch.nn中存在的实际模块，就像简单地执行torch.cuda会指向{{1} }，因此将无法正常工作。

是否有一种方法可以告诉torch._C.cuda在我做pylint时或什至在看到火炬时都看torch和torch._C？我认为我不会使用import torch来引用代码中的任何其他内容。

Answer 1

目前的解决方案是将torch添加到generated-members：

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

或在pylintrc部分的[TYPECHECK]中：

generated-members=torch.*

我在关于pytorch问题reply的github讨论的[Minor Bug] Pylint E1101 Module 'torch' has no 'from_numpy' member #701中找到了这个解决方案。不如白名单令人满意，因为我想如果您引用的东西实际上不是会员，那么它就不会起作用，但这是我到目前为止遇到的最好的解决方案。