PyCharm 预期类型警告

Question

我的代码包含以下几行：

import numpy as np
T = 1.00
tlist = np.linspace(0, T, 1000)

当使用 tlist 作为函数的输入时，我收到以下警告：

尝试通过指定类型来解决这个问题

tlist : np.ndarray = np.linspace(0, T, 1000)

结果是：

有没有办法在不禁用类型检查的情况下解决这个问题？

Answer 1

如果您的函数定义如下所示：

def my_function(tlist):

您可以修改它以包含输入参数 tlist 的类型注释，如下所示：

from typing import Union
import numpy as np
def my_function(tlist: Union[np.ndarray, tuple[np.ndarray, float]]):