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

Question

我正在使用pytorch并且pylint无法识别 ex 的一些功能：torch.stack但是，如果我这样做import torch._C as torch，它似乎工作正常。

如果我在上面这样做，那么存在于torch包中的实际模块就像torch.cuda或者torch.nn需要单独导入一样简单地torch.cuda指向torch._C.cuda并且因此将不起作用。

有没有办法告诉我pylint同时看两个torch，torch._C什么时候看import torch，甚至什么时候看到火炬？我认为我不会torch在我的代码中引用任何其他内容。

标签： pythonimportpytorchpylint

Answer 1

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

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

或在pylintrc以下[TYPECHECK]部分：

generated-members=torch.*

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