python - Mypy 错误 "Need more than 2 values to unpack (3 expected)"尽管使用了不同元组的并集

Question

当我对以下代码运行 mypy 时，我得到了

Need more than 2 values to unpack (3 expected)

我该如何解决这个问题？

def func(third: bool = False) -> tuple[int, int] | tuple[int, int, int]:
    if third:
        return 1, 2, 3
    else:
        return 1, 2
        
a, b, c = func(True) # mypy error line

func 在这种情况下实际上返回了三个值。我正在使用 Python 3.10.4。

Answer 1

描述的情况是 overloading 的典型用例.

您可以注释该函数，使其对于 True 参数返回 3 元组，对于 False - 2 元组:

from typing import overload, Literal

@overload
def func(third: Literal[True]) -> tuple[int, int, int]: 
    ...
@overload
def func(third: Literal[False] = ...) -> tuple[int, int]:
    ...
    
def func(third: bool = False) -> tuple[int, int] | tuple[int, int, int]:
    if third:
        return 1, 2, 3
    else:
        return 1, 2
        
a, b, c = func(True)
a, b = func(False)
a, b = func()

Playground

您当前代码中的问题是 mypy 不知道返回类型是 2 元组还是 3 元组。它不会对实现做出额外的假设，类型提示是严格的。您的代码可能如下所示(由 mypy 以完全相同的方式解释):

import random
def func(three: bool = False) -> tuple[int, int] | tuple[int, int, int]:
    if random.random() < 0.5:
        return 1, 1
    else:
        return 1, 1, 1

a, b, c = func()

...现在您的代码在(大约)50% 的情况下失败并且 mypy 指向该错误。