使用元组列表重塑数组时出现 mypy 错误

Question

我有一个如下所示的 python 模块：

import numpy as np

def test(order: str = 'C') -> np.ndarray:
    return np.array([1,2,3]).reshape((1,2), order = order)

用 mypy 评估它会返回错误和注释

error: No overload variant of "reshape" of "_ArrayOrScalarCommon" matches argument types "Tuple[int, int]", "str"
note: Possible overload variants:
note:     def reshape(self, Sequence[int], *, order: Union[Literal['A'], Literal['C'], Literal['F'], None] = ...) -> ndarray
note:     def reshape(self, *shape: int, order: Union[Literal['A'], Literal['C'], Literal['F'], None] = ...) -> ndarray

将函数调用中的部分 order = order 替换为 order = 'C' 可以防止发生此错误。如果我选择将此参数作为函数参数传递，为什么 mypy 会出现问题？

Answer 1

test(order: str = 'C')只设置参数C的默认值 但您仍然可以使用任何str 调用它。例如test('X') 是 正确输入。但这会导致test 调用reshape order='X' 这是错误的，这就是 mypy 抱怨的原因，这是 好东西。

所以问题出在order的类型定义上。你可以改变 它与reshape 签名匹配，mypy 对此很满意：

import typing as tp
import numpy as np

def test(order: tp.Optional[tp.Literal['A', 'C', 'F']] = 'C') -> np.ndarray:
    return np.array([1,2,3]).reshape((1,2), order = order)

test('C')  # ok
test('X')  # mypy error