PyCharm预期类型警告

jhkqcmku 于 2023-05-17 发布在 PyCharm

关注(0)|答案(1)|浏览(214)

我的代码包含以下行：

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

当使用tlist作为函数的输入时，我得到以下警告：Expected type 'Type[ndarray]', got 'Union[ndarray, tuple[ndarray, Optional[float]]]' instead
尝试通过指定类型tlist : np.ndarray = np.linspace(0, T, 1000)来修复此问题会导致以下结果：Expected type 'Type[ndarray]', got 'Union[ndarray, ndarray]' instead
有没有办法在不禁用类型检查的情况下解决这个问题？

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