Python类型注解报错：List[Type[A]]无法适配List[Union[Type[test],Type[C]]]

问题代码 #

class A:
    pass

class B:
    pass

class C:
    pass

from typing import TypeVar, Union, Type, Optional, List

test = TypeVar("test", bound=A)

def wow(w: Optional[List[Union[Type["test"], Type[C]]]]):
    pass

wow([A]);

问题描述 #

上述代码中，Pylance将wow([A])标记为错误。移除Union部分后代码可正常运行，疑惑添加Union为何导致报错，需要明确是代码编写问题还是Pylance的问题。

错误信息 #

Argument of type "list[Type[A]]" cannot be assigned to parameter "w" of type "List[Type[test@wow] | Type[C]] | None" in function "wow"
  Type "list[Type[A]]" cannot be assigned to type "List[Type[test@wow] | Type[C]] | None"
    "list[Type[A]]" is incompatible with "List[Type[test@wow] | Type[C]]"
      TypeVar "_T@list" is invariant
        Type "Type[A]" cannot be assigned to type "Type[test@wow] | Type[C]"
          "Type[A]" is incompatible with "Type[C]"
          Type "Type[A]" cannot be assigned to type "Type[C]"
    Type cannot be assigned to type "None"

原因分析 #

1. 未绑定的TypeVar：你定义的test = TypeVar("test", bound=A)是泛型类型变量，但wow函数并未声明为泛型函数。在Union[Type["test"], Type[C]]中使用未绑定的TypeVar时，Pylance无法正确推断其具体类型，导致Type[test]无法与Type[A]兼容。
2. 列表类型的不变性：Python类型系统中，列表是不变类型，即list[Type[A]]不能直接赋值给List[Union[Type[X], Type[C]]]——前者元素类型是更具体的Type[A]，而后者是联合类型，不变类型要求元素类型完全匹配。

修复方案 #

方案1：将函数声明为泛型函数 #

如果需要支持A的子类类型或C的类型，把wow定义为泛型函数，明确TypeVar的作用：

class A:
    pass

class B:
    pass

class C:
    pass

from typing import TypeVar, Union, Type, Optional, List

TestT = TypeVar("TestT", bound=A)

def wow(w: Optional[List[Union[Type[TestT], Type[C]]]]) -> None:
    pass

wow([A])  # 不再报错

方案2：直接使用具体类型 #

如果不需要泛型，直接用Type[A]替代Type[test]即可：

class A:
    pass

class B:
    pass

class C:
    pass

from typing import Union, Type, Optional, List

def wow(w: Optional[List[Union[Type[A], Type[C]]]]) -> None:
    pass

wow([A])  # 正常运行

补充说明 #

移除Union后代码正常的原因：此时参数类型为Optional[List[Type[test]]]，Pylance会自动将test推断为A，与传入的[A]类型匹配。但添加Union后，未绑定的TypeVar无法与联合类型完成兼容匹配，因此触发报错。

内容的提问来源于stack exchange，提问作者Planete