如何解决来自pyright的“类型部分未知”警告？

我通过pyright 使用严格的类型检查。

当我有一个返回 pytorch 的方法时DataLoader，然后 Pyright 抱怨我的类型定义：

声明的返回类型“DataLoader[Unknown]”部分未知 Pyright（报告未知变量类型）

查看 pytorch 的类型存根DataLoader（精简到重要部分）：

class DataLoader(Generic[T_co]):
    dataset: Dataset[T_co]

    @overload
    def __init__(self, dataset: Dataset[T_co], ...

据我所知，泛型类型T_co of the DataLoader应由以下定义__init__数据集参数。

Pyright 也抱怨我的Dataset类型定义：

参数“数据集”的类型部分未知 参数类型为“数据集[未知]” Pyright（报告未知参数类型）

看看Dataset类型存根：

class Dataset(Generic[T_co]):
    def __getitem__(self, index: int) -> T_co: ...

向我表明该类型应该由返回类型推断__getitem__.

我的数据集的类型签名__getitem__看起来像这样：

def __getitem__(self, index: int) -> Tuple[Tensor, Tensor]:

基于此我期望Dataset and DataLoader可以推断为Dataset[Tuple[Tensor, Tensor]] and DataLoader[Tuple[Tensor, Tensor]]但事实并非如此。

我的猜测是 Pyright 无法静态推断这里的类型。

我想我可以像这样定义自己的类型签名：

Dataset[Tuple[Tensor, Tensor]]

但这实际上导致我的 python 脚本崩溃：

类型错误：“类型”对象不可下标

我怎样才能正确定义类型Dataset and DataLoader?

由于这个问题没有得到答复，我不确定这是否真的是pyright中的一个错误。因此我在 github 存储库上打开了这个问题：https://github.com/microsoft/pyright/issues/698 https://github.com/microsoft/pyright/issues/698

Eric Traut 详细解释了问题所在，并表示 Pyright 正在按设计运行。 我试图在这里给出要点的要点。

问题解释

如果未提供返回类型，Pyright 会尝试推断返回类型，但如果像本例一样提供了返回类型，则需要对它们进行完全类型化。 Pyright 不会填充给定类型注释的缺失部分。

例如，pyright 将尝试推断以下函数定义的返回类型：

def get_dataset():

但如果返回类型指定为Dataset那么这就是 Pyright 期望的返回类型。

def get_dataset() -> Dataset:

在这种情况下Dataset是一个通用类，不处理像这样的下标Dataset[int]。 在 Python 3.7（我们正在使用的）中，Python 解释器将评估这些类型注释，从而导致上述异常。

Solution

从 Python 3.10 开始，Python 解释器将不再评估类型注释，并且以下类型注释将起作用：

def get_dataset() -> Dataset[int]:

从 Python 3.7 开始，可以通过以下导入启用此行为：

from __future__ import annotations

这记录在PEP 563 https://www.python.org/dev/peps/pep-0563/。 您还需要禁用 pylint 的规则 E1136，以不发出有关“不可订阅对象”的警告。

另一种解决方法是引用类型定义，如下所示：

def get_dataset() -> "Dataset[int]":