Python 2 中的 Python 3 类型提示

Question

我有 python def 定义似乎适用于 python3：

def get_default_device(use_gpu: bool = True) -> cl.Device:

在python2下出现以下语法错误：

Traceback (most recent call last):
  File "map_copy.py", line 9, in <module>
    import utility
  File "/home/root/pyopencla/ch3/utility.py", line 6
    def get_default_device(use_gpu: bool = True) -> cl.Device:
                                  ^
SyntaxError: invalid syntax

如何使类型提示与 python2 兼容？

Answer 1

PEP 3107 中为 Python 3.0 引入了函数注释。在 Python 3.5+ 的 PEP 484 中正式将注解用作类型提示。

3.0 之前的任何版本都将完全不支持您用于类型提示的语法。但是，PEP 484 offers a workaround，一些编辑可能会选择兑现。在您的情况下，提示将如下所示：

def get_default_device(use_gpu=True):
    # type: (bool) -> cl.Device
    ...

或更详细地说，

def get_default_device(use_gpu=True  # type: bool
                      ):
    # type: (...) -> cl.Device
    ...

PEP 明确指出，这种形式的类型提示应该适用于任何版本的 Python，如果它完全受支持的话。