python-Mypy：具有函数字典的不兼容类型(对象)

发布时间：2020-12-17 17:37:22 所属栏目：Python 来源：网络整理

导读：如下代码： from typing import Uniondef a() - int: return 1def b() - str: return 'a'def c(arg: str = 'a') - Union[int,str]: return {'a': a,'b': b}[arg]() 触发以下mypy异常： error: Incompatible return value type (got "object",expected "Union

如下代码：

from typing import Union

def a() -> int:
    return 1

def b() -> str:
    return 'a'

def c(arg: str = 'a') -> Union[int,str]:
    return {'a': a,'b': b}[arg]()

触发以下mypy异常：

error: Incompatible return value type (got "object",expected "Union[int,str]")

一种解决方法是使用：

return a() if arg == 'a' else b()

在这种情况下,Mypy不会抱怨,但是如果有两个以上的函数,则字典语法仍然很有用.
有没有解决的办法,还是Mypy错误？

最佳答案

我认为问题在于您没有声明字典的允许类型.尽管在您的代码中很清楚字典中只有两种类型的输出,但是从打字的角度来看,没有什么可以阻止向其添加另一个函数d()的.

您可以尝试以下解决此问题的方法：

from typing import Union,Dict,Callable

output_dictionary : Dict[str,Union[Callable[[],int],Callable[[],str]]] = {'a': a,'b': b}

def c(arg: str = 'a') -> Union[int,str]:
    return output_dictionary[arg]()

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!