Python任意元组解包类型提示问题：PyLance报错的正确写法

在Python 3.9中定义了特定元组类型TypeA、TypeB，通过TypeVar绑定类型变量T后编写的yielding_function，期望接收指定元组类型的可迭代对象并返回同类型迭代器，但触发了PyLance类型错误。

原代码 #

# python3.9
TypeA = tuple[int, float, int, str]
TypeB = tuple[int, str, set]
T = TypeVar("T", TypeA, TypeB)


def f(x: int) -> int:
    return x * x


def yielding_function(payload: Iterable[T]) -> Iterator[T]:
    for x0, *other in payload:
        yield f(x0), *other

PyLance错误信息 #

Expression of type "tuple[int, *tuple[float* | int* | str* | set[Unknown]*, ...]]" cannot be assigned to yield type "T@yielding_function"
  Type "tuple[int, *tuple[float* | int* | str* | set[Unknown]*, ...]]" cannot be assigned to type "T@yielding_function"

问题 #

请问yielding_function的正确类型提示是什么？或有没有更健壮的定义方式？要求输入与输出保持相同类型，不能使用联合类型。

解决方案 #

错误原因 #

PyLance无法自动推导(f(x0), *other)的具体类型与T完全匹配——other被推断为多种类型的联合元组，无法和TypeA/TypeB的剩余元素类型精准对应，因此抛出类型不匹配错误。

方案1：使用cast显式指定类型（适配Python 3.9原生环境） #

通过cast告诉类型检查器，生成的元组就是T类型，直接绕过类型推导的模糊性：

from typing import Iterable, Iterator, TypeVar, cast

TypeA = tuple[int, float, int, str]
TypeB = tuple[int, str, set]
T = TypeVar("T", TypeA, TypeB)


def f(x: int) -> int:
    return x * x


def yielding_function(payload: Iterable[T]) -> Iterator[T]:
    for x0, *other in payload:
        yield cast(T, (f(x0), *other))

方案2：定义精确的元组类型约束（更健壮，需typing_extensions） #

如果可以安装typing_extensions依赖，利用TypeVarTuple和Unpack来约束元组结构：第一个元素必须是int，剩余元素为任意类型。这种方式能让类型检查器自动识别返回类型与输入类型一致，且支持所有符合该结构的元组，不限于TypeA和TypeB：

from typing import Iterable, Iterator, TypeVar
from typing_extensions import TypeVarTuple, Unpack

# 定义元组的剩余元素类型变量
T = TypeVarTuple('T')
# 约束输入元组的第一个元素是int，后面跟着任意类型的元素
TupleType = tuple[int, Unpack[T]]
# 定义协变的类型变量，绑定到上述元组类型
T_co = TypeVar('T_co', bound=TupleType)


def f(x: int) -> int:
    return x * x


def yielding_function(payload: Iterable[T_co]) -> Iterator[T_co]:
    for x0, *other in payload:
        yield (f(x0), *other)

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