Python不导入第三方包能否通过前向引用完成类型提示

你的猜想是正确的，你完全可以仅通过前向引用做类型提示，而不在运行时导入numpy这类仅类型标注用到的依赖，避免冗余导入。针对你遇到的单独检查demo2.py时mypy报错的问题，有行业通用的规范解决方案：

推荐方案：使用条件类型导入 #

Python标准库typing模块提供了TYPE_CHECKING常量，这个常量仅在静态类型检查阶段（比如mypy运行时）值为True，程序实际运行时恒为False。你可以把仅类型标注用到的导入放在if TYPE_CHECKING的分支下，既可以满足mypy的类型解析需求，也不会在运行时执行导入，不会产生额外开销，同时pylint、flake8等代码检查工具也不会将这类导入标记为未使用的冗余导入。

修改后的demo2.py写法如下：

# demo2.py
from typing import TYPE_CHECKING

if TYPE_CHECKING:
    import numpy as np

# Python 3.7+ 可以添加下面的future导入，就不需要用字符串包裹类型标注
# from __future__ import annotations
def my_func(arr: "np.ndarray") -> int:
    return arr[0]

添加后再单独对demo2.py运行mypy就不会再报名称未定义的错误了。

为什么之前的写法会报错 #

即使你用了字符串形式的前向引用，mypy做静态检查时依然需要解析字符串中类型标注对应的实际定义，单独检查demo2.py时找不到np的定义，自然会抛出错误。条件类型导入刚好可以在不影响运行时的前提下，给mypy提供需要的类型定义信息。

可选兜底方案 #

如果你确实不想哪怕是条件导入numpy，可以在类型标注行添加mypy忽略注释：

def my_func(arr: "np.ndarray") -> int: # type: ignore[name-defined]
    return arr[0]

这种方案会跳过对应位置的名称检查，不推荐优先使用。

内容的提问来源于stack exchange，提问作者adam.hendry