0

我试图了解如何使用TypeVarpyright。

我构建了以下小功能：

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])

[版权：reportGeneralTypeIssues]

此外，它还抱怨：

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

因为：

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

和：

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

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

谢谢！

4

1 回答 1

-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
于 2021-06-01T02:32:30.803 回答