Python 2 中的 Python 3 类型提示

Question

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

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

在 python2 下，我收到以下语法错误：

root:~/pyopencla/ch3# python map_copy.py
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

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

3.0 之前的任何版本都将根本不支持您用于类型提示的语法。 但是，PEP 484 提供了一种解决方法，一些编辑可能会选择尊重它。 在您的情况下，提示如下所示：

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，如果它受支持的话。