Pylance对两种列表联合类型的意外类型推断问题

这不是Pylance的bug，而是你对泛型不变性和类型推断逻辑的理解存在偏差。

核心原因：Set的类型变量是不变的 #

Python标准库中set的定义使用了不变类型变量（TypeVar('_T', covariant=False, contravariant=False)）。这意味着：

• set[int | str]既不能被视为set[int]的子类，也不能被视为set[str]的子类，反之亦然。
• 当你传入list[int] | list[str]类型的参数x时，类型检查器会将x的类型"合并"为list[int | str]（联合类型的泛型容器会被向上合并），因此set(x)会被推断为set[int | str]，而非set[int] | set[str]。

解决方法 #

要让类型检查器正确推断出set[int] | set[str]，可以通过以下两种方式：

1. 使用泛型函数

通过泛型明确绑定输入列表的元素类型，让返回值的集合类型与输入列表的元素类型严格对应：

from typing import TypeVar, List, Set

T = TypeVar('T', int, str)

def my_set(x: List[T]) -> Set[T]:
    return set(x)

2. 使用类型守卫

通过显式的类型判断，将x的类型窄化为list[int]或list[str]，从而让set(x)的类型被正确推断：

def my_set(x: list[int] | list[str]) -> set[int] | set[str]:
    if x and isinstance(x[0], int):
        return set(x)  # 此时x被窄化为list[int]，返回set[int]
    else:
        return set(x)  # 此时x被窄化为list[str]，返回set[str]

补充说明 #

联合类型的泛型容器处理是类型检查器的常规逻辑：当传入一个联合类型的容器时，类型检查器无法在运行前确定具体是哪一种子类型，因此会采用最宽泛的推断结果（合并元素类型后的容器类型）。而由于set的不变性，这个宽泛的类型无法与你声明的返回类型（两个具体集合类型的联合）兼容，从而触发报错。

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