以下代码:
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 不会抱怨,但如果有超过 2 个函数,字典语法仍然很有帮助。有没有办法解决它,或者它是一个 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]()