mypy推断Python类型签名时出现意外Arg包装及类型匹配问题

问题核心原因 #

• 泛型参数重复定义：OwnIterable类继承Generic[T]后，内部又重复定义了T = TypeVar("T")，这导致类的泛型参数和外层DemoClass使用的T是两个完全独立的类型变量。OwnIterable.__iter__返回的Iterator[set[T]]里的T是类内部的新变量，和inner_fun参数的set[T]不匹配——这才是MyPy报错的根本原因。而Arg(Set[T], 'C')只是MyPy显示带参数名的类型签名时的内部格式，并非类型不兼容的核心问题。
• frozenset与set的类型差异：使用inner_fun_frozen时，frozenset[T]和OwnIterable返回的set[T]是明确的不同类型，MyPy直接显示类型不匹配，不会触发Arg格式的显示（类型差异更直接）。
• 单行语句的额外错误：把map和sum合并成单行时，MyPy的类型推断链因为前面的泛型冲突出了问题，误将sum的参数推断为需要Iterable[bool]（类似filter的返回值类型），所以额外弹出了期望Callable[[Set[T]], bool]的错误，这是类型推断失败后的次生问题。

解决方案 #

修复OwnIterable里的泛型参数定义，删除内部重复的TypeVar声明：

from __future__ import annotations
from typing import Generic, Iterator, TypeVar

T = TypeVar("T")

class OwnIterable(Generic[T]):
    # 移除这里重复定义的T = TypeVar("T")
    def __iter__(self) -> Iterator[set[T]]:
        return [].__iter__()

class DemoClass(Generic[T]):
    def __call__(self, P: OwnIterable[T]) -> float:
        def inner_fun(C: set[T]) -> float:
            return 42.0

        def inner_fun_frozen(C: frozenset[T]) -> float:
            return 42.0

        map_obj = map(inner_fun, P)
        a = sum(map_obj)

        # 这里的错误是预期内的：frozenset[T]和set[T]类型不兼容
        map_obj_frozen = map(inner_fun_frozen, P)
        b = sum(map_obj_frozen)

        c = sum(map(inner_fun, P))

        return a + b + c

补充说明 #

• 修复泛型参数后，map(inner_fun, P)的类型错误会消失，因为两者的set[T]现在使用的是同一个泛型参数T。
• inner_fun_frozen的错误是正常的：frozenset和set是不可互相转换的不同类型。如果需要兼容，可以把inner_fun_frozen的参数改成Iterable[T]（业务逻辑允许的话），或者把OwnIterable的返回类型改成Iterator[frozenset[T]]。
• 单行语句的额外错误也会随着泛型参数的修复而消失，因为类型推断链恢复了正常。

内容的提问来源于stack exchange，提问作者D. A.