pyright：在具有最小值/最大值的齐次列表中使用 TypeVar

Question

我尝试了解如何将

TypeVar

与pyright 一起使用。

我构建了以下小功能：

import random
from typing import List, Sequence, TypeVar


T = TypeVar("T", int, str, float)
TypedList = List[T]


def merge(a: TypedList, b: TypedList) -> TypedList:
    i0, i1 = 0, 0
    merged = []

    # Run with two pointers
    while i0 < len(a) and i1 < len(b):
        merged.append(min(a[i0], b[i1]))
        if merged[-1] == a[i0]:
            i0 += 1
        else:
            i1 += 1

    # Add the tail
    if i0 >= len(a):
        merged.extend(b[i1:])
    if i1 >= len(b):
        merged.extend(a[i0:])

    return merged

让pyright在线投诉：

    merged.append(min(a[i0], b[i1]))

因为：

“min”没有重载与提供的参数匹配

参数类型：(TypedList[int], TypedList[int])

[Pyright：报告一般类型问题]

此外，它还抱怨：

if merged[-1] == a[i0]

因为：

预期的类类型，但收到“int”

和：

非法类型注释：不允许使用变量，除非它是类型别名

mypy 似乎在这条线上工作得很好。 关于问题是什么以及如何解决它有什么想法吗？

谢谢！

Answer 1

使用“Union”而不是“TypeVar”

import random
from typing import List, Sequence, TypeVar, Union


U = Union[int, str, float]
TypedList = List[U]


def merge(a: TypedList, b: TypedList) -> TypedList:
    i0, i1 = 0, 0
    merged = []

    # Run with two pointers
    while i0 < len(a) and i1 < len(b):
        merged.append(min(a[i0], b[i1]))
        if merged[-1] == a[i0]:
            i0 += 1
        else:
            i1 += 1

    # Add the tail
    if i0 >= len(a):
        merged.extend(b[i1:])
    if i1 >= len(b):
        merged.extend(a[i0:])

    return merged

在使用'TypeVar'时，如'T = TypeVar(...)'，所有T应该是相同的类型。所以，pyright已经很困惑了。在这种情况下，您应该使用“Union”，因为“U”不一定是同一类型。

或者像这样

import random
from typing import List, Sequence, TypeVar


T = TypeVar("T", int, str, float)


def merge(a: list[T], b: list[T]) -> list[T]:
    i0, i1 = 0, 0
    merged = []

    # Run with two pointers
    while i0 < len(a) and i1 < len(b):
        merged.append(min(a[i0], b[i1]))
        if merged[-1] == a[i0]:
            i0 += 1
        else:
            i1 += 1

    # Add the tail
    if i0 >= len(a):
        merged.extend(b[i1:])
    if i1 >= len(b):
        merged.extend(a[i0:])

    return merged