Google Colab 在多大程度上支持 Python 打字？

Question

我输入了包含这些行的代码。

from typing import Dict, List, Set, Tuple

def pairs_sum_to_k(a_set: Set[int], k: int) -> List[Tuple[int, int]]:
    ...

代码已编译并运行。那挺好的。当我尝试导入

typing

中没有的内容时，Colab 生成了一条错误消息，这也很好。

不好的是，当类型提示与程序不一致时，例如将返回类型更改为简单的

int

，Colab 没有抱怨。这表明 Colab 可以处理类型提示语法，但它对类型声明根本不做任何事情。是这样吗？我应该从 Colab 获得什么样的输入支持（如果有）？

谢谢。

Answer 1

Python 中的类型注释只是装饰——Python 本身不进行任何类型验证。来自 Python 文档：

注意： Python 运行时不强制执行函数和变量类型注释。它们可以被第三方工具使用，例如类型检查器、IDE、linter 等。

如果你想验证你的类型，你需要使用像 mypy 这样的工具，它就是为此而设计的。

我不知道 Colab 中有任何内置类型检查功能，但自己定义相对简单。例如，您可以创建一个 Jupyter cell magic，使用 mypy 对单元格内容执行类型检查：

# Simple mypy cell magic for Colab
!pip install mypy
from IPython.core.magic import register_cell_magic
from IPython import get_ipython
from mypy import api

@register_cell_magic
def mypy(line, cell):
  for output in api.run(['-c', '\n' + cell] + line.split()):
    if output and not output.startswith('Success'):
      raise TypeError(output)
  get_ipython().run_cell(cell)

然后你可以像这样使用它：

%%mypy

def foo(x: int) -> int:
  return 2 * x

foo('a')

执行后，这是输出：

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-6-21dcff84b262> in <module>()
----> 1 get_ipython().run_cell_magic('mypy', '', "\ndef foo(x: int) -> int:\n  return 2 * x\n\nfoo('a')")

/usr/local/lib/python3.6/dist-packages/IPython/core/interactiveshell.py in run_cell_magic(self, magic_name, line, cell)
   2115             magic_arg_s = self.var_expand(line, stack_depth)
   2116             with self.builtin_trap:
-> 2117                 result = fn(magic_arg_s, cell)
   2118             return result
   2119 

<ipython-input-5-d2e45a31f6bb> in mypy(line, cell)
      8   for output in api.run(['-c', '\n' + cell] + line.split()):
      9     if output:
---> 10       raise TypeError(output)
     11   get_ipython().run_cell(cell)

TypeError: <string>:6: error: Argument 1 to "foo" has incompatible type "str"; expected "int"
Found 1 error in 1 file (checked 1 source file)

Answer 2

您可以使用 nb-mypy 在 google colab 中进行类型检查。这将在每个后续单元执行时自动运行类型检查。

# install nb-mypy (suppress output)
!pip install nb-mypy -qqq
# load extension
%load_ext nb_mypy

如果您使用第 3 方库，您还需要安装库特定类型以进行有效的类型检查（请参阅 PEP561）。例如，要安装 dateutil 的类型：

!pip install types-python-dateutil -qqq