结合 mypy Union 和嵌套的 TypedDict 导致 mypy errror: incompatible return value

Question

我想将TypedDict与Union结合起来。 这样一个函数就可以返回A或B 。 Mypy 能够直接正确检测TypedDict返回类型。 但是当TypedDict嵌套在Union它不起作用。

from typing_extensions import TypedDict
from typing import Union


class A(TypedDict):
    a: str


class B(TypedDict):
    b: str


def works() -> A:
    return {'a': 'value'}
    # Works as expected


def problem() -> Union[A, B]:
    return {'a': 'value'}
    # mypy_error: Incompatible return value type (got "Dict[str, str]", expected "Union[A, B]")
    # Reports an error while it should be valid


def workaround() -> Union[A, B]:
    x: A = {'a': 'value'}

    return x
    # This works again but is not very elegant

一种可能的解决方法是分配给临时返回的类型提示变量（请参阅workaround() ）。 有没有更优雅的方法来做到这一点？

注意：Python 3.7

Answer 1

引用自PEP 589 ：

通常需要显式的 [TypedDict] 类型注释，否则类型检查器可以假设普通字典类型，以实现向后兼容。 当类型检查器可以推断构造的字典对象应该是 TypedDict 时，可以省略显式注释。

因此，在代码中明确定义类型并没有错。 另一种可能性是直接“实例化” A ：

def problem() -> Union[A, B]:
    return A(a='value')

虽然这当然只是一个语法糖，在运行时会被dict替换。