python - Pylint 与 pytorch : Is there a way to tell pylint to look for module in different place?

Question

我正在使用pytorch并且pylint无法识别例如torch.stack的几个函数，但是，如果我导入torch._C as torch 似乎工作正常。

如果我执行上述操作，torch 包中存在的实际模块(如 torch.cuda 或 torch.nn)需要单独导入，如下所示:执行 torch.cuda 会指向 torch._C.cuda，因此不起作用。

当我导入torch时，有没有办法告诉pylint同时查看torch和torch._C code> 或者甚至每当它看到 torch 时？我认为我不会使用 torch 来引用代码中的任何其他内容。

Answer 1

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

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

或者在[TYPECHECK]部分下的pylintrc中:

generated-members=torch.*

我在 reply 中找到了这个解决方案到pytorch问题的github讨论[Minor Bug] Pylint E1101 Module 'torch' has no 'from_numpy' member #701 。不如白名单令人满意，因为我想如果您引用实际上不是成员的内容，它不会被捕获，但这是迄今为止我遇到的最佳解决方案。