spaCy 的 python 类型提示

Question

我正在努力为涉及 spaCy 类型的任何内容设置类型提示。

尝试使用 spaCy 文档中的“语言”和“令牌”等返回类型，但它们似乎不起作用......我在 python 3.8.5 上。

例如说我有这个：

import spacy

def load_lang() -> what_goes_here:
   return spacy.load("en_core_web_sm")

根据文档，这个“返回语言”。 我尝试了 Language 或 spacy.Language 但没有运气。 我在使用令牌时遇到类似的问题，例如做这样的事情：

return [token for token in doc if not token.is_stop]

文档再次提到“返回令牌”作为返回 object 但我无法输入提示。 我不想 go 从打字到 List[str] 下，因为我假设这是 go 的错误方式。

任何帮助都感激不尽。 先感谢您。

Answer 1

如果您使用以下命令导入Language class 应该没有问题：

from spacy.language import Language

然后将其用作类型提示：

def load_lang() -> Language:
   return spacy.load("en_core_web_sm")