泛型函数返回语句触发mypy类型检查错误的原因咨询

我来帮你拆解一下这个问题哈～首先先把你的代码和遇到的报错再明确一下：

你写的代码是：

def incr[T: int | None](value: T) -> T:
    if value is None:
        return value
    return value + 1

incr(None)
incr(1)

运行mypy后得到的错误是：

main.py:4: error: Incompatible return value type (got "int", expected "T") [return-value]
Found 1 error in 1 file (checked 1 source file)

为什么会出现这个错误？ #

问题出在你定义泛型类型变量的方式上。你用的[T: int | None]是给类型变量T设置了上界约束，意思是T必须是int | None的子类型，但这种写法没办法让mypy建立起「输入类型和返回类型的精确关联」——虽然你在代码里已经通过if value is None分支处理了None的情况，但mypy没办法从这个上界约束推断出：当进入return value +1分支时，T一定是int类型。

从mypy的类型检查逻辑来看，它只知道T是int | None的子类型，但不确定在当前分支里T具体是什么，所以它会认为你返回的int可能和函数声明的返回类型T不匹配（比如理论上T还可能是None，虽然你的代码逻辑已经排除了这种情况，但泛型的类型约束没给mypy足够的信息）。

怎么解决这个问题？ #

你的需求是「输入None返回None，输入int返回int」，这种情况需要用受限类型变量来定义T，也就是明确告诉mypy：T只能是int或者None这两个类型之一，这样mypy就能准确推断每个分支的返回类型和输入类型一致。

修改后的代码如下：

from typing import TypeVar

# 定义只能是int或None的类型变量T
T = TypeVar('T', int, None)

def incr(value: T) -> T:
    if value is None:
        return value
    return value + 1

incr(None)
incr(1)

现在再运行mypy就不会报错了，因为这个写法把T的可能值限定死了，mypy能清晰地对应：当输入是None时，返回None；当输入是int时，返回int，完全符合函数声明的T -> T的类型约定。

备注：内容来源于stack exchange，提问作者袛屑懈褌褉懈泄 袙芯褉芯薪械褑泻懈泄