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")