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