Mypy“类模块”注释

Question

我有一个函数load_config，它可以加载并返回一个python模块：

import imp

def load_config(path: str):
    return imp.load_source('config', path)

print(type(load_config('config.py')))

此代码段打印<class 'module'>。

如何使用Mypy注释load_config返回值？

Answer 1

正确的注释是：

import imp
import types

def load_config(path: str) -> types.ModuleType:
  return imp.load_source('config', path)

但是，有一个open issue可以在当前版本的mypy中修复它。